Update pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -859,6 +859,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 859 |
image = torch.cat([image] * 2)
|
| 860 |
|
| 861 |
print("prepared control image_batch_size", image.shape)
|
|
|
|
| 862 |
|
| 863 |
return image
|
| 864 |
|
|
|
|
| 859 |
image = torch.cat([image] * 2)
|
| 860 |
|
| 861 |
print("prepared control image_batch_size", image.shape)
|
| 862 |
+
print("prepared control device", image.device)
|
| 863 |
|
| 864 |
return image
|
| 865 |
|