KingNish commited on
Commit
87dfbfe
Β·
verified Β·
1 Parent(s): 5c7c14b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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_V4.0_Lightning", torch_dtype=torch.float16, vae=vae)
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 = """
@@ -92,6 +90,8 @@ def king(type ,
92
  generator = torch.Generator().manual_seed(seed)
93
  if fast:
94
  pipes=pipe_fast
 
 
95
  else:
96
  pipes=pipe
97
  image = pipes( prompt = instruction,
 
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 = """
 
90
  generator = torch.Generator().manual_seed(seed)
91
  if fast:
92
  pipes=pipe_fast
93
+ steps=int(steps/2)
94
+ guidance_scale=(guidance_scale/3)
95
  else:
96
  pipes=pipe
97
  image = pipes( prompt = instruction,