Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,8 +70,6 @@ def generate(
|
|
| 70 |
torch_dtype=torch.float16,
|
| 71 |
use_safetensors=True,
|
| 72 |
)
|
| 73 |
-
|
| 74 |
-
pipe.load_lora_weights(lora)
|
| 75 |
|
| 76 |
if ENABLE_CPU_OFFLOAD:
|
| 77 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 70 |
torch_dtype=torch.float16,
|
| 71 |
use_safetensors=True,
|
| 72 |
)
|
|
|
|
|
|
|
| 73 |
|
| 74 |
if ENABLE_CPU_OFFLOAD:
|
| 75 |
pipe.enable_model_cpu_offload()
|