Commit
·
06e49f5
1
Parent(s):
1491344
test padding mode
Browse files- handler.py +7 -2
handler.py
CHANGED
@@ -100,6 +100,12 @@ class EndpointHandler():
|
|
100 |
self.pipe.controlnet = self.controlnet
|
101 |
|
102 |
|
|
|
|
|
|
|
|
|
|
|
|
|
103 |
# hyperparamters
|
104 |
negatice_prompt = data.pop("negative_prompt", None)
|
105 |
num_inference_steps = data.pop("num_inference_steps", 30)
|
@@ -131,8 +137,7 @@ class EndpointHandler():
|
|
131 |
height=height,
|
132 |
width=width,
|
133 |
controlnet_conditioning_scale=controlnet_conditioning_scale,
|
134 |
-
generator=self.generator
|
135 |
-
tiling=tiling
|
136 |
)
|
137 |
|
138 |
|
|
|
100 |
self.pipe.controlnet = self.controlnet
|
101 |
|
102 |
|
103 |
+
targets = [self.pipe.vae, self.pipe.unet]
|
104 |
+
for target in targets:
|
105 |
+
for module in target.modules():
|
106 |
+
if isinstance(module, torch.nn.Conv2d):
|
107 |
+
module.padding_mode = "circular"
|
108 |
+
|
109 |
# hyperparamters
|
110 |
negatice_prompt = data.pop("negative_prompt", None)
|
111 |
num_inference_steps = data.pop("num_inference_steps", 30)
|
|
|
137 |
height=height,
|
138 |
width=width,
|
139 |
controlnet_conditioning_scale=controlnet_conditioning_scale,
|
140 |
+
generator=self.generator
|
|
|
141 |
)
|
142 |
|
143 |
|