camenduru commited on
Commit
2845655
·
verified ·
1 Parent(s): c3d1a06

Update worker_runpod.py

Browse files
Files changed (1) hide show
  1. worker_runpod.py +2 -0
worker_runpod.py CHANGED
@@ -21,12 +21,14 @@ def generate(input):
21
  flow_shift = values['flow_shift']
22
  batch_size = values['batch_size']
23
  embedded_guidance_scale = values['embedded_guidance_scale']
 
24
 
25
  if seed == 0:
26
  random.seed(int(time.time()))
27
  seed = random.randint(0, 18446744073709551615)
28
 
29
  args = parse_args()
 
30
  hunyuan_video_sampler = HunyuanVideoSampler.from_pretrained("/content/HunyuanVideo/ckpts", args=args)
31
  args = hunyuan_video_sampler.args
32
 
 
21
  flow_shift = values['flow_shift']
22
  batch_size = values['batch_size']
23
  embedded_guidance_scale = values['embedded_guidance_scale']
24
+ flow_reverse = values['flow_reverse']
25
 
26
  if seed == 0:
27
  random.seed(int(time.time()))
28
  seed = random.randint(0, 18446744073709551615)
29
 
30
  args = parse_args()
31
+ args.flow_reverse = flow_reverse
32
  hunyuan_video_sampler = HunyuanVideoSampler.from_pretrained("/content/HunyuanVideo/ckpts", args=args)
33
  args = hunyuan_video_sampler.args
34