Update pipeline.py
Browse files- pipeline.py +3 -0
pipeline.py
CHANGED
|
@@ -1138,6 +1138,9 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 1138 |
batch_size = len(prompt)
|
| 1139 |
else:
|
| 1140 |
batch_size = prompt_embeds.shape[0]
|
|
|
|
|
|
|
|
|
|
| 1141 |
|
| 1142 |
device = self._execution_device
|
| 1143 |
|
|
|
|
| 1138 |
batch_size = len(prompt)
|
| 1139 |
else:
|
| 1140 |
batch_size = prompt_embeds.shape[0]
|
| 1141 |
+
|
| 1142 |
+
# print promtp embed shape
|
| 1143 |
+
print(prompt_embeds.shape)
|
| 1144 |
|
| 1145 |
device = self._execution_device
|
| 1146 |
|