yamildiego commited on
Commit
2316dce
·
1 Parent(s): a831ddd

add tilling

Browse files
Files changed (1) hide show
  1. handler.py +3 -1
handler.py CHANGED
@@ -110,6 +110,7 @@ class EndpointHandler():
110
  controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1.0)
111
 
112
  test_var = data.pop("test_var", "DEFAULT")
 
113
 
114
  print(f"prompt: {prompt}")
115
  print(f"prompt: {test_var}")
@@ -130,7 +131,8 @@ class EndpointHandler():
130
  height=height,
131
  width=width,
132
  controlnet_conditioning_scale=controlnet_conditioning_scale,
133
- generator=self.generator
 
134
  )
135
 
136
 
 
110
  controlnet_conditioning_scale = data.pop("controlnet_conditioning_scale", 1.0)
111
 
112
  test_var = data.pop("test_var", "DEFAULT")
113
+ tiling = data.pop("tiling", False)
114
 
115
  print(f"prompt: {prompt}")
116
  print(f"prompt: {test_var}")
 
131
  height=height,
132
  width=width,
133
  controlnet_conditioning_scale=controlnet_conditioning_scale,
134
+ generator=self.generator,
135
+ tiling=tiling
136
  )
137
 
138