jonathanpark commited on
Commit
20b727c
1 Parent(s): e4d55ce

add additional parameters

Browse files
Files changed (2) hide show
  1. README.md +6 -0
  2. handler.py +1 -1
README.md CHANGED
@@ -7,6 +7,12 @@ widget:
7
  - src: >-
8
  https://huggingface.co/datasets/mishig/sample_images/resolve/main/canny-edge.jpg
9
  prompt: Girl with Pearl Earring
 
 
 
 
 
 
10
  duplicated_from: stabilityai/stable-diffusion-2-1-base
11
  ---
12
 
 
7
  - src: >-
8
  https://huggingface.co/datasets/mishig/sample_images/resolve/main/canny-edge.jpg
9
  prompt: Girl with Pearl Earring
10
+ num_inference_steps: ''
11
+ guidance_scale: ''
12
+ negative_prompt: ''
13
+ height: ''
14
+ width: ''
15
+ manual_seed: ''
16
  duplicated_from: stabilityai/stable-diffusion-2-1-base
17
  ---
18
 
handler.py CHANGED
@@ -51,7 +51,7 @@ class EndpointHandler():
51
 
52
  out = None
53
 
54
- if data.get("url"):
55
  generator = torch.Generator(device='cuda')
56
  generator.manual_seed(manual_seed)
57
  # run img2img pipeline
 
51
 
52
  out = None
53
 
54
+ if data.get("src"):
55
  generator = torch.Generator(device='cuda')
56
  generator.manual_seed(manual_seed)
57
  # run img2img pipeline