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]
|