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

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +1 -1
generate.py CHANGED
@@ -189,7 +189,7 @@ def generate(args):
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"{args.task}_{args.size}_{args.ulysses_size}_{args.ring_size}_{formatted_prompt}_{formatted_time}" + suffix
193
 
194
  try:
195
  if "t2i" in args.task:
 
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: