Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -1137,7 +1137,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
| 1137 |
elif prompt is not None and isinstance(prompt, list):
|
| 1138 |
batch_size = len(prompt)
|
| 1139 |
else:
|
| 1140 |
-
batch_size =
|
| 1141 |
|
| 1142 |
# print promtp embed shape
|
| 1143 |
print(prompt_embeds.shape)
|
|
|
|
| 1137 |
elif prompt is not None and isinstance(prompt, list):
|
| 1138 |
batch_size = len(prompt)
|
| 1139 |
else:
|
| 1140 |
+
batch_size = 1
|
| 1141 |
|
| 1142 |
# print promtp embed shape
|
| 1143 |
print(prompt_embeds.shape)
|