Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,13 +61,11 @@ def generate(
|
|
| 61 |
|
| 62 |
if not use_vae:
|
| 63 |
pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
|
| 64 |
-
|
| 65 |
-
|
| 66 |
if use_vae:
|
| 67 |
vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
|
| 68 |
pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
|
| 69 |
-
|
| 70 |
-
|
| 71 |
if use_lora:
|
| 72 |
pipe.load_lora_weights(lora)
|
| 73 |
pipe.fuse_lora(lora_scale=0.7)
|
|
|
|
| 61 |
|
| 62 |
if not use_vae:
|
| 63 |
pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch.float16)
|
| 64 |
+
|
|
|
|
| 65 |
if use_vae:
|
| 66 |
vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
|
| 67 |
pipe = DiffusionPipeline.from_pretrained(model, vae=vae, torch_dtype=torch.float16)
|
| 68 |
+
|
|
|
|
| 69 |
if use_lora:
|
| 70 |
pipe.load_lora_weights(lora)
|
| 71 |
pipe.fuse_lora(lora_scale=0.7)
|