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",
|