smzerbe commited on
Commit
0693744
·
verified ·
1 Parent(s): de9be62

Update modules/sd_models.py

Browse files
Files changed (1) hide show
  1. 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 devices.autocast(), torch.no_grad():
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")