ar0551 commited on
Commit
caee6ce
·
verified ·
1 Parent(s): 911d75e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -32,7 +32,11 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
32
  vae=vae,
33
  torch_dtype=precision,
34
  scheduler=eulera_scheduler,
35
- ).to(device)
 
 
 
 
36
 
37
  # 📸 Edge detection function using OpenCV (Canny)
38
  @spaces.GPU
 
32
  vae=vae,
33
  torch_dtype=precision,
34
  scheduler=eulera_scheduler,
35
+ )
36
+
37
+ # Load IKEA lora
38
+ pipe.load_lora_weights("ostris/ikea-instructions-lora-sdxl")
39
+ pipe.to(device)
40
 
41
  # 📸 Edge detection function using OpenCV (Canny)
42
  @spaces.GPU