Spaces:
Running
on
Zero
Running
on
Zero
Update raw.py
Browse files
raw.py
CHANGED
@@ -22,7 +22,7 @@ pipe.to("cuda")
|
|
22 |
@spaces.GPU
|
23 |
def generate_image(prompt, control_image):
|
24 |
# Load control image
|
25 |
-
control_image = load_image(control_image
|
26 |
w, h = control_image.size
|
27 |
# Upscale x4
|
28 |
control_image = control_image.resize((w * 2, h * 2))
|
@@ -30,7 +30,7 @@ def generate_image(prompt, control_image):
|
|
30 |
prompt=prompt,
|
31 |
control_image=control_image,
|
32 |
controlnet_conditioning_scale=0.6,
|
33 |
-
num_inference_steps=
|
34 |
guidance_scale=3.5,
|
35 |
height=control_image.size[1],
|
36 |
width=control_image.size[0]
|
|
|
22 |
@spaces.GPU
|
23 |
def generate_image(prompt, control_image):
|
24 |
# Load control image
|
25 |
+
control_image = load_image(control_image)
|
26 |
w, h = control_image.size
|
27 |
# Upscale x4
|
28 |
control_image = control_image.resize((w * 2, h * 2))
|
|
|
30 |
prompt=prompt,
|
31 |
control_image=control_image,
|
32 |
controlnet_conditioning_scale=0.6,
|
33 |
+
num_inference_steps=8,
|
34 |
guidance_scale=3.5,
|
35 |
height=control_image.size[1],
|
36 |
width=control_image.size[0]
|