Udate config
Browse files- 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
|
|
|
|
| 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
|