KingNish commited on
Commit
535533b
·
verified ·
1 Parent(s): 7a8e778

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -29,8 +29,9 @@ if torch.cuda.is_available():
29
  add_watermarker=False
30
  )
31
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
32
- pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
33
- pipe.set_adapters("dalle")
 
34
 
35
  pipe.to("cuda")
36
 
 
29
  add_watermarker=False
30
  )
31
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
32
+ pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle2")
33
+ pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle1")
34
+ pipe.set_adapters(["dalle1", "dalle2"], adapter_weights=[1.0, 0.5])
35
 
36
  pipe.to("cuda")
37