asoderznik
/

sdx4-upscaler

StableDiffusionUpscalePipeline

stable-diffusion

Inference Endpoints

Model card Files Files and versions Community

asoderznik commited on Aug 29, 2023

Commit

6915ef8

•

1 Parent(s): 96b1d9e

try something

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -22,7 +22,8 @@ if device.type != 'cuda':
 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
-        self.pipe = StableDiffusionPipeline.from_pretrained(path, torch_dtype=torch.float16)
         self.pipe = self.pipe.to(device)

 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
+        model_id = "stabilityai/stable-diffusion-x4-upscaler"
+        self.pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
         self.pipe = self.pipe.to(device)