karimbenharrak commited on
Commit
2c7e90a
1 Parent(s): ed1bd2d

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +6 -6
handler.py CHANGED
@@ -111,9 +111,9 @@ class EndpointHandler():
111
  negative_prompt=negative_prompt,
112
  image=image,
113
  mask_image=mask_image,
114
- guidance_scale=guidance_scale, #8.0
115
- num_inference_steps=num_inference_steps, #100
116
- strength=strength, #0.2
117
  output_type="latent", # let's keep in latent to save some VRAM
118
  ).images[0]
119
 
@@ -124,9 +124,9 @@ class EndpointHandler():
124
  image2 = self.pipe3(
125
  prompt=prompt,
126
  image=image,
127
- guidance_scale=guidance_scale, #8.0
128
- num_inference_steps=num_inference_steps, #100
129
- strength=strength, #0.2
130
  ).images[0]
131
 
132
  print("3rd pipeline part successful!")
 
111
  negative_prompt=negative_prompt,
112
  image=image,
113
  mask_image=mask_image,
114
+ guidance_scale=8.0, #8.0
115
+ num_inference_steps=50, #100
116
+ strength=0.2, #0.2
117
  output_type="latent", # let's keep in latent to save some VRAM
118
  ).images[0]
119
 
 
124
  image2 = self.pipe3(
125
  prompt=prompt,
126
  image=image,
127
+ guidance_scale=8.0, #8.0
128
+ num_inference_steps=50, #100
129
+ strength=0.2, #0.2
130
  ).images[0]
131
 
132
  print("3rd pipeline part successful!")