Update server/pipelines/pix2pixTurbo.py
Browse files
server/pipelines/pix2pixTurbo.py
CHANGED
@@ -49,23 +49,6 @@ class Pipeline:
|
|
49 |
height: int = Field(
|
50 |
512, min=2, max=15, title="Height", disabled=True, hide=True, id="height"
|
51 |
)
|
52 |
-
strength: float = Field(
|
53 |
-
1.0,
|
54 |
-
min=0.01,
|
55 |
-
max=10.0,
|
56 |
-
step=0.001,
|
57 |
-
title="Strength",
|
58 |
-
field="range",
|
59 |
-
hide=True,
|
60 |
-
id="strength",
|
61 |
-
)
|
62 |
-
deterministic: bool = Field(
|
63 |
-
True,
|
64 |
-
hide=True,
|
65 |
-
title="Deterministic",
|
66 |
-
field="checkbox",
|
67 |
-
id="deterministic",
|
68 |
-
)
|
69 |
canny_low_threshold: float = Field(
|
70 |
0.0,
|
71 |
min=0,
|
@@ -111,8 +94,6 @@ class Pipeline:
|
|
111 |
output_image = self.model(
|
112 |
canny_tensor,
|
113 |
params.prompt,
|
114 |
-
params.deterministic,
|
115 |
-
params.strength,
|
116 |
)
|
117 |
output_pil = transforms.ToPILImage()(output_image[0].cpu() * 0.5 + 0.5)
|
118 |
|
|
|
49 |
height: int = Field(
|
50 |
512, min=2, max=15, title="Height", disabled=True, hide=True, id="height"
|
51 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
52 |
canny_low_threshold: float = Field(
|
53 |
0.0,
|
54 |
min=0,
|
|
|
94 |
output_image = self.model(
|
95 |
canny_tensor,
|
96 |
params.prompt,
|
|
|
|
|
97 |
)
|
98 |
output_pil = transforms.ToPILImage()(output_image[0].cpu() * 0.5 + 0.5)
|
99 |
|