Elvenson
/

diffuser_inference

stable-diffusion

stable-diffusion-diffusers

endpoints-template

Model card Files Files and versions Community

Quoc Bao Bui commited on Dec 3, 2022

Commit

eddf5bd

·

1 Parent(s): e762c44

Add access token

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -20,7 +20,7 @@ def decode_base64_image(image_string):
 class EndpointHandler:
     def __init__(self, path=""):
         self.pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16,
-            revision="fp16")
         self.pipe = self.pipe.to("cuda")
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:

 class EndpointHandler:
     def __init__(self, path=""):
         self.pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16,
+            revision="fp16", use_auth_token="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")
         self.pipe = self.pipe.to("cuda")
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]: