ameerazam08 commited on
Commit
3aa4a89
·
verified ·
1 Parent(s): 5f6eaf4

refiner setup

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,7 +32,7 @@ CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0")
32
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
33
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
34
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
35
- ENABLE_REFINER = os.getenv("ENABLE_REFINER", "0") == "1"
36
 
37
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
38
  if torch.cuda.is_available():
@@ -219,14 +219,14 @@ with gr.Blocks(css="footer{display:none !important}", theme=theme) as demo:
219
  minimum=1,
220
  maximum=20,
221
  step=0.1,
222
- value=5.0,
223
  )
224
  num_inference_steps_base = gr.Slider(
225
  label="Number of inference steps for base",
226
  minimum=10,
227
  maximum=100,
228
  step=1,
229
- value=25,
230
  )
231
  with gr.Row(visible=False) as refiner_params:
232
  guidance_scale_refiner = gr.Slider(
 
32
  MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
33
  USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
34
  ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD") == "1"
35
+ ENABLE_REFINER = os.getenv("ENABLE_REFINER", "0")#
36
 
37
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
38
  if torch.cuda.is_available():
 
219
  minimum=1,
220
  maximum=20,
221
  step=0.1,
222
+ value=7.5,
223
  )
224
  num_inference_steps_base = gr.Slider(
225
  label="Number of inference steps for base",
226
  minimum=10,
227
  maximum=100,
228
  step=1,
229
+ value=60,
230
  )
231
  with gr.Row(visible=False) as refiner_params:
232
  guidance_scale_refiner = gr.Slider(