Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 =
|
89 |
-
num_inference_steps_refiner: int =
|
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:
|