surajx commited on
Commit
b66f429
·
verified ·
1 Parent(s): 6198b8a

removed `variant=fp16`

Browse files
Files changed (1) hide show
  1. 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", use_safetensors=True
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(