Update inference.py
Browse files- inference.py +1 -0
inference.py
CHANGED
|
@@ -76,6 +76,7 @@ pipe = HunyuanVideoPipeline.from_pretrained(model_id, transformer=transformer, t
|
|
| 76 |
# Enable memory savings
|
| 77 |
pipe.vae.enable_tiling()
|
| 78 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 79 |
|
| 80 |
with torch.no_grad(): # enable image inputs
|
| 81 |
initial_input_channels = pipe.transformer.config.in_channels
|
|
|
|
| 76 |
# Enable memory savings
|
| 77 |
pipe.vae.enable_tiling()
|
| 78 |
pipe.enable_model_cpu_offload()
|
| 79 |
+
pipe.enable_model_cpu_offload()
|
| 80 |
|
| 81 |
with torch.no_grad(): # enable image inputs
|
| 82 |
initial_input_channels = pipe.transformer.config.in_channels
|