hysts HF Staff commited on
Commit
f0e8e9b
·
1 Parent(s): c1bad21
Files changed (1) hide show
  1. app.py +8 -10
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
- if ENABLE_REFINER:
30
- refiner = DiffusionPipeline.from_pretrained(
31
- "stabilityai/stable-diffusion-xl-refiner-1.0",
32
- vae=vae,
33
- torch_dtype=torch.float16,
34
- use_safetensors=True,
35
- variant="fp16",
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, visible=ENABLE_REFINER)
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",