SunderAli17 commited on
Commit
6e18009
·
verified ·
1 Parent(s): 90ea810

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,7 +38,7 @@ if torch.cuda.is_available():
38
  "dataautogpt3/ProteusV0.2",
39
  vae=vae,
40
  torch_dtype=torch.float16,
41
- use_safetensors=True,
42
  # variant="fp16",
43
  )
44
  if ENABLE_REFINER:
@@ -85,8 +85,8 @@ def infer(
85
  height: int = 1024,
86
  guidance_scale_base: float = 5.0,
87
  guidance_scale_refiner: float = 5.0,
88
- num_inference_steps_base: int = 25,
89
- num_inference_steps_refiner: int = 25,
90
  apply_refiner: bool = False,
91
  progress=gr.Progress(track_tqdm=True),
92
  ) -> PIL.Image.Image:
 
38
  "dataautogpt3/ProteusV0.2",
39
  vae=vae,
40
  torch_dtype=torch.float16,
41
+ # use_safetensors=True,
42
  # variant="fp16",
43
  )
44
  if ENABLE_REFINER:
 
85
  height: int = 1024,
86
  guidance_scale_base: float = 5.0,
87
  guidance_scale_refiner: float = 5.0,
88
+ num_inference_steps_base: int = 30,
89
+ num_inference_steps_refiner: int = 30,
90
  apply_refiner: bool = False,
91
  progress=gr.Progress(track_tqdm=True),
92
  ) -> PIL.Image.Image: