Commit
·
3efa11b
1
Parent(s):
b81ee75
no control net
Browse files- handler.py +6 -7
handler.py
CHANGED
@@ -92,12 +92,12 @@ class EndpointHandler():
|
|
92 |
return {"error": "Please provide a prompt and base64 encoded image."}
|
93 |
|
94 |
# Check if a new controlnet is provided
|
95 |
-
if controlnet_type is not None and controlnet_type != self.control_type:
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
101 |
|
102 |
|
103 |
targets = [self.pipe.vae, self.pipe.unet]
|
@@ -107,7 +107,6 @@ class EndpointHandler():
|
|
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)
|
112 |
guidance_scale = data.pop("guidance_scale", 7.4)
|
113 |
negative_prompt = data.pop("negative_prompt", None)
|
|
|
92 |
return {"error": "Please provide a prompt and base64 encoded image."}
|
93 |
|
94 |
# Check if a new controlnet is provided
|
95 |
+
#if controlnet_type is not None and controlnet_type != self.control_type:
|
96 |
+
# print(f"changing controlnet from {self.control_type} to {controlnet_type} using {CONTROLNET_MAPPING[controlnet_type]['model_id']} model")
|
97 |
+
# self.control_type = controlnet_type
|
98 |
+
# self.controlnet = ControlNetModel.from_pretrained(CONTROLNET_MAPPING[self.control_type]["model_id"],
|
99 |
+
# torch_dtype=dtype).to(device)
|
100 |
+
# self.pipe.controlnet = self.controlnet
|
101 |
|
102 |
|
103 |
targets = [self.pipe.vae, self.pipe.unet]
|
|
|
107 |
module.padding_mode = "circular"
|
108 |
|
109 |
# hyperparamters
|
|
|
110 |
num_inference_steps = data.pop("num_inference_steps", 30)
|
111 |
guidance_scale = data.pop("guidance_scale", 7.4)
|
112 |
negative_prompt = data.pop("negative_prompt", None)
|