asoderznik
/

sdx4-upscaler

StableDiffusionUpscalePipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

asoderznik commited on Aug 31, 2023

Commit

43b0eb9

•

1 Parent(s): 4e4734a

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -35,7 +35,7 @@ class EndpointHandler():
         self.pipe = StableDiffusionUpscalePipeline.from_pretrained(self.path, torch_dtype=torch.float16)
         self.pipe = self.pipe.to(device)
-        self.pipe.enable_xformers_memory_efficient_attention()
         logger.info('data received %s', data)
         inputs = data.get("inputs")
         logger.info('inputs received %s', inputs)

         self.pipe = StableDiffusionUpscalePipeline.from_pretrained(self.path, torch_dtype=torch.float16)
         self.pipe = self.pipe.to(device)
+        #self.pipe.enable_xformers_memory_efficient_attention()
         logger.info('data received %s', data)
         inputs = data.get("inputs")
         logger.info('inputs received %s', inputs)