Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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,
|
| 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/
|
| 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
|