1inkusFace commited on
Commit
42d98b4
·
verified ·
1 Parent(s): 11907e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -30,11 +30,11 @@ def init_predictor(task_type: str):
30
  task_type=TaskType.I2V if task_type == "i2v" else TaskType.T2V,
31
  model_id="Skywork/skyreels-v1-Hunyuan-i2v", # Adjust model ID as needed
32
  quant_model=True,
33
- is_offload=True, # Consider removing if you have enough GPU memory
34
- offload_config=OffloadConfig(
35
- high_cpu_memory=True,
36
- parameters_level=True,
37
- ),
38
  use_multiprocessing=False,
39
  )
40
  return predictor
 
30
  task_type=TaskType.I2V if task_type == "i2v" else TaskType.T2V,
31
  model_id="Skywork/skyreels-v1-Hunyuan-i2v", # Adjust model ID as needed
32
  quant_model=True,
33
+ is_offload=False, # Consider removing if you have enough GPU memory
34
+ offload_config=None, #OffloadConfig(
35
+ # high_cpu_memory=True,
36
+ # parameters_level=True,
37
+ #),
38
  use_multiprocessing=False,
39
  )
40
  return predictor