Spaces:
Runtime error
Runtime error
Update worker_runpod.py
Browse files- 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 |
|