removed `variant=fp16`
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -17,7 +17,7 @@ class EndpointHandler:
|
|
| 17 |
def __init__(self, path=""):
|
| 18 |
# load StableDiffusionInpaintPipeline pipeline
|
| 19 |
self.base = DiffusionPipeline.from_pretrained(
|
| 20 |
-
path, torch_dtype=torch.float16, variant="fp16"
|
| 21 |
)
|
| 22 |
# use DPMSolverMultistepScheduler
|
| 23 |
self.base.scheduler = DPMSolverMultistepScheduler.from_config(
|
|
@@ -33,7 +33,7 @@ class EndpointHandler:
|
|
| 33 |
vae=self.base.vae,
|
| 34 |
torch_dtype=torch.float16,
|
| 35 |
use_safetensors=True,
|
| 36 |
-
variant="fp16",
|
| 37 |
)
|
| 38 |
# use DPMSolverMultistepScheduler
|
| 39 |
self.refiner.scheduler = DPMSolverMultistepScheduler.from_config(
|
|
|
|
| 17 |
def __init__(self, path=""):
|
| 18 |
# load StableDiffusionInpaintPipeline pipeline
|
| 19 |
self.base = DiffusionPipeline.from_pretrained(
|
| 20 |
+
path, torch_dtype=torch.float16, use_safetensors=True # variant="fp16"
|
| 21 |
)
|
| 22 |
# use DPMSolverMultistepScheduler
|
| 23 |
self.base.scheduler = DPMSolverMultistepScheduler.from_config(
|
|
|
|
| 33 |
vae=self.base.vae,
|
| 34 |
torch_dtype=torch.float16,
|
| 35 |
use_safetensors=True,
|
| 36 |
+
# variant="fp16",
|
| 37 |
)
|
| 38 |
# use DPMSolverMultistepScheduler
|
| 39 |
self.refiner.scheduler = DPMSolverMultistepScheduler.from_config(
|