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
|