KingNish commited on
Commit
35f15b4
Β·
verified Β·
1 Parent(s): 909646e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -18,9 +18,7 @@ pipe.to("cuda")
18
  refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
19
  refiner.to("cuda")
20
 
21
- pipe_fast = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V3.0_Turbo", torch_dtype=torch.float16, vae=vae, use_safetensors=True, variant="fp16")
22
- pipe_fast.load_lora_weights("KingNish/Better-Image-XL-Lora", weight_name="example-03.safetensors", adapter_name="lora")
23
- pipe_fast.set_adapters("lora")
24
  pipe_fast.to("cuda")
25
 
26
  help_text = """
@@ -93,7 +91,7 @@ def king(type ,
93
  generator = torch.Generator().manual_seed(seed)
94
  if fast:
95
  pipes=pipe_fast
96
- steps=int(steps/3)
97
  guidance_scale=(guidance_scale/3)
98
  else:
99
  pipes=pipe
 
18
  refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", vae=vae, torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
19
  refiner.to("cuda")
20
 
21
+ pipe_fast = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V3.0_Turbo", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
 
 
22
  pipe_fast.to("cuda")
23
 
24
  help_text = """
 
91
  generator = torch.Generator().manual_seed(seed)
92
  if fast:
93
  pipes=pipe_fast
94
+ steps=int(steps/2)
95
  guidance_scale=(guidance_scale/3)
96
  else:
97
  pipes=pipe