ds1david commited on
Commit
160e39b
Β·
1 Parent(s): 5779b8d

fixing bugs

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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