Upload folder using huggingface_hub

handler.py

@@ -32,7 +32,7 @@ class EndpointHandler:
         if self.pipe is None:
             del self.pipe
             torch.cuda.empty_cache()
-            self.pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=torch.float16, variant="fp16").to("cuda")
+            self.pipe = StableDiffusionXLPipeline.from_pretrained(base).to("cuda")
 
             # Ensure sampler uses "trailing" timesteps.
             self.pipe.scheduler = DPMSolverSinglestepScheduler.from_config(self.pipe.scheduler.config, timestep_spacing="trailing")