working endpoint with pwV5
Browse files- handler.py +8 -7
- text_encoder/pytorch_model.bin +1 -1
- unet/diffusion_pytorch_model.bin +1 -1
- vae/diffusion_pytorch_model.bin +1 -1
handler.py
CHANGED
@@ -103,18 +103,19 @@ class EndpointHandler:
|
|
103 |
|
104 |
# DPM++ 2M SDE Karras
|
105 |
# increase step to avoid high contrast num_inference_steps=30
|
106 |
-
# self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
107 |
-
# self.pipe.scheduler.config,
|
108 |
-
# use_karras_sigmas=True,
|
109 |
-
# algorithm_type="sde-dpmsolver++",
|
110 |
-
# )
|
111 |
-
# DPM++ 2M Karras
|
112 |
-
# increase step to avoid high contrast num_inference_steps=30
|
113 |
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
114 |
self.pipe.scheduler.config,
|
115 |
use_karras_sigmas=True,
|
|
|
116 |
)
|
117 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
118 |
# Mode boulardus
|
119 |
self.pipe.safety_checker = None
|
120 |
|
|
|
103 |
|
104 |
# DPM++ 2M SDE Karras
|
105 |
# increase step to avoid high contrast num_inference_steps=30
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
106 |
self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
107 |
self.pipe.scheduler.config,
|
108 |
use_karras_sigmas=True,
|
109 |
+
algorithm_type="sde-dpmsolver++",
|
110 |
)
|
111 |
|
112 |
+
# DPM++ 2M Karras
|
113 |
+
# increase step to avoid high contrast num_inference_steps=30
|
114 |
+
# self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(
|
115 |
+
# self.pipe.scheduler.config,
|
116 |
+
# use_karras_sigmas=True,
|
117 |
+
# )
|
118 |
+
|
119 |
# Mode boulardus
|
120 |
self.pipe.safety_checker = None
|
121 |
|
text_encoder/pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 492306077
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:41844a711a087b43222654f31e6035c98215e25b041834331282e9288edb11be
|
3 |
size 492306077
|
unet/diffusion_pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 3438366373
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e00ae441399298118f144e556f9517042faa352fdb4410a70f28c739ae4ee25d
|
3 |
size 3438366373
|
vae/diffusion_pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 334712113
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1b01618945554d9840701d3453d4a9fe3db0db090164a5ed6305641306285b6f
|
3 |
size 334712113
|