prithivMLmods commited on
Commit
5d23a57
·
verified ·
1 Parent(s): da938bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,12 +39,12 @@ ENABLE_CPU_OFFLOAD = 0
39
 
40
  if torch.cuda.is_available():
41
  pipe = StableDiffusionXLPipeline.from_pretrained(
42
- "fluently/Fluently-XL-v4",
43
  torch_dtype=torch.float16,
44
  use_safetensors=True,
45
  )
46
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
47
- pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
48
  pipe.set_adapters("dalle")
49
  pipe.to("cuda")
50
 
 
39
 
40
  if torch.cuda.is_available():
41
  pipe = StableDiffusionXLPipeline.from_pretrained(
42
+ "prithivMLmods/NGC5128-StableDiffusion-XL-v4",
43
  torch_dtype=torch.float16,
44
  use_safetensors=True,
45
  )
46
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
47
+ pipe.load_lora_weights("prithivMLmods/Pinwheel-DallE-3-Lora-Xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
48
  pipe.set_adapters("dalle")
49
  pipe.to("cuda")
50