fixing bugs
Browse files
app.py
CHANGED
|
@@ -85,14 +85,14 @@ try:
|
|
| 85 |
pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
| 86 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
| 87 |
torch_dtype=torch_dtype,
|
| 88 |
-
use_safetensors=True
|
| 89 |
-
variant=None # ForΓ§ar carregamento sem fp16
|
| 90 |
).to(device)
|
| 91 |
|
| 92 |
# LoRA
|
| 93 |
pipe.load_lora_weights(
|
| 94 |
"KappaNeuro/bas-relief",
|
| 95 |
-
weight_name="BAS-RELIEF.safetensors"
|
|
|
|
| 96 |
)
|
| 97 |
|
| 98 |
# Modelo de profundidade
|
|
|
|
| 85 |
pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
| 86 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
| 87 |
torch_dtype=torch_dtype,
|
| 88 |
+
use_safetensors=True
|
|
|
|
| 89 |
).to(device)
|
| 90 |
|
| 91 |
# LoRA
|
| 92 |
pipe.load_lora_weights(
|
| 93 |
"KappaNeuro/bas-relief",
|
| 94 |
+
weight_name="BAS-RELIEF.safetensors",
|
| 95 |
+
peft_backend="peft" # This is crucial
|
| 96 |
)
|
| 97 |
|
| 98 |
# Modelo de profundidade
|