Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
app.py
CHANGED
@@ -31,8 +31,9 @@ controlnet = ControlNetModel.from_pretrained(
|
|
31 |
)
|
32 |
|
33 |
controlnet_lineart = ControlNetModel.from_pretrained(
|
34 |
-
"
|
35 |
-
torch_dtype=torch.float16
|
|
|
36 |
)
|
37 |
|
38 |
def check_use_custom_or_no(value):
|
@@ -149,13 +150,6 @@ def infer(use_custom_model, model_name, weight_name, custom_lora_weight, image_i
|
|
149 |
image = np.concatenate([image, image, image], axis=2)
|
150 |
image = Image.fromarray(image)
|
151 |
|
152 |
-
if preprocessor == "lineart":
|
153 |
-
image = load_image(image_in)
|
154 |
-
image = np.array(image)
|
155 |
-
image = HWC3(image)
|
156 |
-
#image = resize_image(image, resolution=image_resolution)
|
157 |
-
image = Image.fromarray(image)
|
158 |
-
|
159 |
if use_custom_model:
|
160 |
|
161 |
if model_name == "":
|
|
|
31 |
)
|
32 |
|
33 |
controlnet_lineart = ControlNetModel.from_pretrained(
|
34 |
+
"TheMistoAI/MistoLine",
|
35 |
+
torch_dtype=torch.float16,
|
36 |
+
variant="fp16",
|
37 |
)
|
38 |
|
39 |
def check_use_custom_or_no(value):
|
|
|
150 |
image = np.concatenate([image, image, image], axis=2)
|
151 |
image = Image.fromarray(image)
|
152 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
153 |
if use_custom_model:
|
154 |
|
155 |
if model_name == "":
|