SaitoHoujou commited on
Commit
7d140d1
·
verified ·
1 Parent(s): 64dcc99

Udate config

Browse files
Files changed (1) hide show
  1. src/config.py +1 -1
src/config.py CHANGED
@@ -6,4 +6,4 @@ app_controlnet_path = "SaitoHoujou/Finetuned-ControlNet_for_Diffusion-Model"
6
  app_base_model = "botp/stable-diffusion-v1-5"
7
  app_hed_model = 'lllyasviel/Annotators'
8
  app_device = "cuda" if torch.cuda.is_available() else "cpu"
9
- app_dtype = torch.float16 if device == "cuda" else torch.float32
 
6
  app_base_model = "botp/stable-diffusion-v1-5"
7
  app_hed_model = 'lllyasviel/Annotators'
8
  app_device = "cuda" if torch.cuda.is_available() else "cpu"
9
+ app_dtype = torch.float16 if app_device == "cuda" else torch.float32