ar0551 commited on
Commit
666d6cd
·
verified ·
1 Parent(s): 461fe92

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -41,12 +41,15 @@ pipe.disable_lora()
41
 
42
  def activate_ikea_lora():
43
  pipe.set_adapters("ikea")
 
44
 
45
  def activate_pixel_lora():
46
  pipe.set_adapters("pixel")
 
47
 
48
  def disable_loras():
49
  pipe.disable_lora()
 
50
 
51
  pipe.to(device)
52
 
 
41
 
42
  def activate_ikea_lora():
43
  pipe.set_adapters("ikea")
44
+ pipe.to(device)
45
 
46
  def activate_pixel_lora():
47
  pipe.set_adapters("pixel")
48
+ pipe.to(device)
49
 
50
  def disable_loras():
51
  pipe.disable_lora()
52
+ pipe.to(device)
53
 
54
  pipe.to(device)
55