1inkusFace commited on
Commit
c657f20
·
verified ·
1 Parent(s): caa59ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -20,7 +20,7 @@ def init_predictor():
20
  predictor = SkyReelsVideoSingleGpuInfer(
21
  task_type= TaskType.I2V,
22
  model_id="Skywork/SkyReels-V1-Hunyuan-I2V",
23
- quant_model=True,
24
  is_offload=False,
25
  offload_config=OffloadConfig(
26
  high_cpu_memory=True,
@@ -37,10 +37,10 @@ def generate_video(prompt, seed, image=None):
37
  seed = int(random.randrange(4294967294))
38
  kwargs = {
39
  "prompt": prompt,
40
- "height": 512,
41
- "width": 512,
42
- "num_frames": 97,
43
- "num_inference_steps": 30,
44
  "seed": seed,
45
  "guidance_scale": 6.0,
46
  "embedded_guidance_scale": 1.0,
 
20
  predictor = SkyReelsVideoSingleGpuInfer(
21
  task_type= TaskType.I2V,
22
  model_id="Skywork/SkyReels-V1-Hunyuan-I2V",
23
+ quant_model=False,
24
  is_offload=False,
25
  offload_config=OffloadConfig(
26
  high_cpu_memory=True,
 
37
  seed = int(random.randrange(4294967294))
38
  kwargs = {
39
  "prompt": prompt,
40
+ "height": 386,
41
+ "width": 386,
42
+ "num_frames": 64,
43
+ "num_inference_steps": 10,
44
  "seed": seed,
45
  "guidance_scale": 6.0,
46
  "embedded_guidance_scale": 1.0,