prithivMLmods commited on
Commit
33fb8d8
1 Parent(s): 39d5854

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -37,12 +37,12 @@ ENABLE_CPU_OFFLOAD = 0
37
 
38
  if torch.cuda.is_available():
39
  pipe = StableDiffusionXLPipeline.from_pretrained(
40
- "stabilityai/stable-diffusion-xl-base-1.0",
41
  torch_dtype=torch.float16,
42
  use_safetensors=True,
43
  )
44
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
45
- pipe.load_lora_weights("h1t/TCD-SDXL-LoRA", weight_name="pytorch_lora_weights.safetensors", adapter_name="dalle")
46
  pipe.set_adapters("dalle")
47
  pipe.to("cuda")
48
 
 
37
 
38
  if torch.cuda.is_available():
39
  pipe = StableDiffusionXLPipeline.from_pretrained(
40
+ "stabilityai/stable-cascade",
41
  torch_dtype=torch.float16,
42
  use_safetensors=True,
43
  )
44
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
45
+ pipe.load_lora_weights("dataautogpt3/OpenDalleV1.1", weight_name="OpenDalleV1.1.safetensors", adapter_name="dalle")
46
  pipe.set_adapters("dalle")
47
  pipe.to("cuda")
48