Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,8 +54,6 @@ def generate(
|
|
| 54 |
) -> PIL.Image.Image:
|
| 55 |
if torch.cuda.is_available():
|
| 56 |
pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
|
| 57 |
-
|
| 58 |
-
pipe.load_lora_weights(lora)
|
| 59 |
|
| 60 |
if ENABLE_CPU_OFFLOAD:
|
| 61 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 54 |
) -> PIL.Image.Image:
|
| 55 |
if torch.cuda.is_available():
|
| 56 |
pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
|
|
|
|
|
|
|
| 57 |
|
| 58 |
if ENABLE_CPU_OFFLOAD:
|
| 59 |
pipe.enable_model_cpu_offload()
|