Wan2.1-sajjad

Running

seokochin commited on 7 days ago

Commit

7949ac3

verified ·

1 Parent(s): 9d7db02

Update generate.py

Files changed (1) hide show

generate.py CHANGED Viewed

@@ -27,7 +27,7 @@ if torch.cuda.is_available():
     torch.backends.cudnn.deterministic = True
 # Run WAN 2.1 - 14B Model
-command = f"python run_model.py --task {args.task} --size {args.size} --frame_num {args.frame_num} --sample_steps {args.sample_steps} --ckpt_dir {args.ckpt_dir} --offload_model {args.offload_model} --prompt \"{args.prompt}\""
 process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 stdout, stderr = process.communicate()

     torch.backends.cudnn.deterministic = True
 # Run WAN 2.1 - 14B Model
+command = f"python generate.py --task {args.task} --size {args.size} --frame_num {args.frame_num} --sample_steps {args.sample_steps} --ckpt_dir {args.ckpt_dir} --offload_model {args.offload_model} --prompt \"{args.prompt}\""
 process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 stdout, stderr = process.communicate()