yamildiego commited on
Commit
66c9a4e
1 Parent(s): 64d3ad8

change documentation

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. handler.py +1 -1
README.md CHANGED
@@ -12,7 +12,7 @@ inference: true
12
 
13
  ### expected Request payload
14
 
15
- **ONLY prompt is mandatory**
16
 
17
  ```json
18
  {
@@ -23,7 +23,7 @@ inference: true
23
  "scale": 7,
24
  "height": 512,
25
  "width": 512,
26
- "controlnet_conditioning_scale": 1,
27
  "sd_model": "dreamshaper"
28
  }
29
  ```
 
12
 
13
  ### expected Request payload
14
 
15
+ **prompt and image_depth_map are mandatory**
16
 
17
  ```json
18
  {
 
23
  "scale": 7,
24
  "height": 512,
25
  "width": 512,
26
+ "controlnet_conditioning_scale": 1.0,
27
  "sd_model": "dreamshaper"
28
  }
29
  ```
handler.py CHANGED
@@ -67,7 +67,7 @@ class EndpointHandler():
67
  scale = data.pop("scale", 7)
68
  height = data.pop("height", None)
69
  width = data.pop("width", None)
70
- controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1)
71
 
72
  # Check if neither prompt nor image is provided
73
  if prompt is None:
 
67
  scale = data.pop("scale", 7)
68
  height = data.pop("height", None)
69
  width = data.pop("width", None)
70
+ controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1.0)
71
 
72
  # Check if neither prompt nor image is provided
73
  if prompt is None: