Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,7 +5,6 @@ from PIL import Image
|
|
| 5 |
import numpy as np
|
| 6 |
import gradio as gr
|
| 7 |
import os
|
| 8 |
-
import gc
|
| 9 |
|
| 10 |
|
| 11 |
model_id = "hunyuanvideo-community/HunyuanVideo"
|
|
@@ -44,7 +43,6 @@ def generate(prompt, width=832, height=832, num_inference_steps=30, lora_id=None
|
|
| 44 |
if lora_id and lora_id.strip() != "":
|
| 45 |
pipe.unload_lora_weights()
|
| 46 |
torch.cuda.empty_cache()
|
| 47 |
-
gc.collect()
|
| 48 |
|
| 49 |
|
| 50 |
iface = gr.Interface(
|
|
|
|
| 5 |
import numpy as np
|
| 6 |
import gradio as gr
|
| 7 |
import os
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
model_id = "hunyuanvideo-community/HunyuanVideo"
|
|
|
|
| 43 |
if lora_id and lora_id.strip() != "":
|
| 44 |
pipe.unload_lora_weights()
|
| 45 |
torch.cuda.empty_cache()
|
|
|
|
| 46 |
|
| 47 |
|
| 48 |
iface = gr.Interface(
|