LPX55 commited on
Commit
57aaf0c
·
verified ·
1 Parent(s): 61b6980

Update raw.py

Browse files
Files changed (1) hide show
  1. raw.py +2 -2
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.name)
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=14,
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]