Update app.py
Browse files
app.py
CHANGED
|
@@ -74,7 +74,8 @@ def get_output_video(text):
|
|
| 74 |
dtype=torch.float16 if fp16 else torch.float32,
|
| 75 |
device=device
|
| 76 |
)
|
| 77 |
-
|
|
|
|
| 78 |
image = model.generate_image(
|
| 79 |
text,
|
| 80 |
seed,
|
|
@@ -83,6 +84,7 @@ def get_output_video(text):
|
|
| 83 |
is_verbose=True
|
| 84 |
)
|
| 85 |
return image
|
|
|
|
| 86 |
|
| 87 |
generated_images = []
|
| 88 |
for senten in plot[:-1]:
|
|
|
|
| 74 |
dtype=torch.float16 if fp16 else torch.float32,
|
| 75 |
device=device
|
| 76 |
)
|
| 77 |
+
|
| 78 |
+
# Ensure correct dtype for inputs
|
| 79 |
image = model.generate_image(
|
| 80 |
text,
|
| 81 |
seed,
|
|
|
|
| 84 |
is_verbose=True
|
| 85 |
)
|
| 86 |
return image
|
| 87 |
+
|
| 88 |
|
| 89 |
generated_images = []
|
| 90 |
for senten in plot[:-1]:
|