Hugging Face's logo

Spaces:
LuyangZ
/
FloorAI
Runtime error

App Files Community
1
LuyangZ commited on
Commit
00c47e9
·
verified ·
1 Parent(s): 928852e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ model_id = "LuyangZ/FloorAI"
23
  # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float16)
24
  # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype="auto")
25
  # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float32)
26
- controlnet = ControlNetModel.from_pretrained(model_id)
27
 
28
  controlnet.to(device)
29
  # torch.cuda.empty_cache()
 
23
  # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float16)
24
  # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype="auto")
25
  # controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float32)
26
+ controlnet = ControlNetModel.from_pretrained(model_id, force_download=True)
27
 
28
  controlnet.to(device)
29
  # torch.cuda.empty_cache()