Spaces:
Running
Running
Update generate.py
Browse files- generate.py +2 -4
generate.py
CHANGED
@@ -186,10 +186,8 @@ def generate(args):
|
|
186 |
# Save the output video or image
|
187 |
if rank == 0:
|
188 |
if args.save_file is None:
|
189 |
-
|
190 |
-
|
191 |
-
suffix = '.png' if "t2i" in args.task else '.mp4'
|
192 |
-
args.save_file = f"examples/video.mp4"
|
193 |
|
194 |
try:
|
195 |
if "t2i" in args.task:
|
|
|
186 |
# Save the output video or image
|
187 |
if rank == 0:
|
188 |
if args.save_file is None:
|
189 |
+
|
190 |
+
args.save_file = f"generated_video.mp4"
|
|
|
|
|
191 |
|
192 |
try:
|
193 |
if "t2i" in args.task:
|