Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
18 |
)
|
19 |
|
20 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
21 |
-
"
|
22 |
controlnet=controlnet,
|
23 |
safety_checker=None,
|
24 |
torch_dtype=torch.float16,
|
@@ -34,9 +34,9 @@ def inference(
|
|
34 |
input_image: Image.Image,
|
35 |
prompt: str,
|
36 |
negative_prompt: str,
|
37 |
-
guidance_scale: float =
|
38 |
-
controlnet_conditioning_scale: float = 1
|
39 |
-
strength: float = 0.
|
40 |
seed: int = -1,
|
41 |
sampler = "DPM++ Karras SDE",
|
42 |
):
|
|
|
18 |
)
|
19 |
|
20 |
pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
|
21 |
+
"SG161222/Realistic_Vision_V2.0",
|
22 |
controlnet=controlnet,
|
23 |
safety_checker=None,
|
24 |
torch_dtype=torch.float16,
|
|
|
34 |
input_image: Image.Image,
|
35 |
prompt: str,
|
36 |
negative_prompt: str,
|
37 |
+
guidance_scale: float = 8.0,
|
38 |
+
controlnet_conditioning_scale: float = 1,
|
39 |
+
strength: float = 0.9,
|
40 |
seed: int = -1,
|
41 |
sampler = "DPM++ Karras SDE",
|
42 |
):
|