rahul7star commited on
Commit
bc1f660
·
verified ·
1 Parent(s): 6ed3768

Update generate.py

Browse files
Files changed (1) hide show
  1. 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
- formatted_time = datetime.now().strftime("%Y%m%d_%H%M%S")
190
- formatted_prompt = args.prompt.replace(" ", "_").replace("/", "_")[:50]
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: