Spaces:
Paused
Paused
Update app.py
Browse files
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=
|
34 |
-
offload_config=OffloadConfig(
|
35 |
-
|
36 |
-
|
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
|