Spaces:
Runtime error
Runtime error
Update modules/sd_models.py
Browse files- modules/sd_models.py +1 -1
modules/sd_models.py
CHANGED
|
@@ -872,7 +872,7 @@ def load_model(checkpoint_info=None, already_loaded_state_dict=None):
|
|
| 872 |
|
| 873 |
timer.record("scripts callbacks")
|
| 874 |
|
| 875 |
-
with
|
| 876 |
sd_model.cond_stage_model_empty_prompt = get_empty_cond(sd_model)
|
| 877 |
|
| 878 |
timer.record("calculate empty prompt")
|
|
|
|
| 872 |
|
| 873 |
timer.record("scripts callbacks")
|
| 874 |
|
| 875 |
+
with torch.no_grad():
|
| 876 |
sd_model.cond_stage_model_empty_prompt = get_empty_cond(sd_model)
|
| 877 |
|
| 878 |
timer.record("calculate empty prompt")
|