Update src/utils.py

src/utils.py

@@ -224,7 +224,7 @@ def rescale_noise_cfg(noise_cfg, noise_pred_text, guidance_rescale=0.0):
     noise_cfg = guidance_rescale * noise_pred_rescaled + (1 - guidance_rescale) * noise_cfg
     return noise_cfg
 
-def register_parallel_pipeline_orig(pipe, mod='50ls'):
+def register_parallel_pipeline_orig(pipe, mod='50ls4'):
     def new_call(self):
         @torch.no_grad()
         def call(
@@ -521,7 +521,7 @@ def register_parallel_pipeline_orig(pipe, mod='50ls'):
 
     pipe.call = new_call(pipe)
 
-def register_faster_forward(model, mod = '50ls'):
+def register_faster_forward(model, mod = '50ls4'):
     def faster_forward(self):
         def forward(
                 sample: torch.FloatTensor,