Update pipeline.py
Browse files- pipeline.py +4 -0
pipeline.py
CHANGED
|
@@ -1428,7 +1428,11 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 1428 |
latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
|
| 1429 |
|
| 1430 |
control_end_step = int(control_end*num_inference_steps)
|
|
|
|
| 1431 |
print("control_end_step", control_end_step)
|
|
|
|
|
|
|
|
|
|
| 1432 |
if self.controlnet != None and i < int(control_end*num_inference_steps):
|
| 1433 |
|
| 1434 |
torch.cuda.synchronize() # Synchronize GPU
|
|
|
|
| 1428 |
latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
|
| 1429 |
|
| 1430 |
control_end_step = int(control_end*num_inference_steps)
|
| 1431 |
+
print(i, control_end_step)
|
| 1432 |
print("control_end_step", control_end_step)
|
| 1433 |
+
if self.controlnet != None and i < control_end_step:
|
| 1434 |
+
print("adding controlnet")
|
| 1435 |
+
|
| 1436 |
if self.controlnet != None and i < int(control_end*num_inference_steps):
|
| 1437 |
|
| 1438 |
torch.cuda.synchronize() # Synchronize GPU
|