Spaces:
Running
on
Zero
Running
on
Zero
Update
Browse files
app.py
CHANGED
@@ -14,7 +14,6 @@ DESCRIPTION = "# SDXL"
|
|
14 |
|
15 |
MAX_SEED = np.iinfo(np.int32).max
|
16 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
|
17 |
-
ENABLE_REFINER = os.getenv("ENABLE_REFINER", "1") == "1"
|
18 |
|
19 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
20 |
|
@@ -26,14 +25,13 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
26 |
use_safetensors=True,
|
27 |
variant="fp16",
|
28 |
).to(device)
|
29 |
-
|
30 |
-
refiner
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
).to(device)
|
37 |
|
38 |
|
39 |
def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
@@ -176,7 +174,7 @@ with gr.Blocks(css_paths="style.css") as demo:
|
|
176 |
step=32,
|
177 |
value=1024,
|
178 |
)
|
179 |
-
apply_refiner = gr.Checkbox(label="Apply refiner", value=False
|
180 |
with gr.Row():
|
181 |
guidance_scale_base = gr.Slider(
|
182 |
label="Guidance scale for base",
|
|
|
14 |
|
15 |
MAX_SEED = np.iinfo(np.int32).max
|
16 |
MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
|
|
|
17 |
|
18 |
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
19 |
|
|
|
25 |
use_safetensors=True,
|
26 |
variant="fp16",
|
27 |
).to(device)
|
28 |
+
refiner = DiffusionPipeline.from_pretrained(
|
29 |
+
"stabilityai/stable-diffusion-xl-refiner-1.0",
|
30 |
+
vae=vae,
|
31 |
+
torch_dtype=torch.float16,
|
32 |
+
use_safetensors=True,
|
33 |
+
variant="fp16",
|
34 |
+
).to(device)
|
|
|
35 |
|
36 |
|
37 |
def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
|
174 |
step=32,
|
175 |
value=1024,
|
176 |
)
|
177 |
+
apply_refiner = gr.Checkbox(label="Apply refiner", value=False)
|
178 |
with gr.Row():
|
179 |
guidance_scale_base = gr.Slider(
|
180 |
label="Guidance scale for base",
|