Hugging Face's logo

Spaces:
LuyangZ
/
FloorAI
Runtime error

App Files Community
1
LuyangZ commited on
Commit
9000286
·
verified ·
1 Parent(s): 1f8c80d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ move_cache()
18
 
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
 
21
- base_model_id = "runwayml/stable-diffusion-v1-5"
 
22
  model_id = "LuyangZ/FloorAI"
23
  # model_id = "LuyangZ/controlnet_Neufert4_64_100"
24
 
 
18
 
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
 
21
+ # base_model_id = "runwayml/stable-diffusion-v1-5"
22
+ base_model_id = "stabilityai/stable-diffusion-2-1-base"
23
  model_id = "LuyangZ/FloorAI"
24
  # model_id = "LuyangZ/controlnet_Neufert4_64_100"
25