Onlyfake
/

realistic-vision-v4

StableDiffusionPipeline

Inference Endpoints

Model card Files Files and versions Community

kapellmeister commited on Jul 18, 2023

Commit

dcac2d5

•

1 Parent(s): 2b90b60

Update handler.py

Files changed (1) hide show

handler.py +2 -3

handler.py CHANGED Viewed

@@ -15,9 +15,8 @@ if device.type != 'cuda':
 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
-        self.pipe = StableDiffusionPipeline.from_single_file(
-        "https://huggingface.co/Onlyfake/realistic-vision-v4/blob/main/Realistic_Vision_V4.0.safetensors"
-)        self.pipe = self.pipe.to(device)
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:

 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
+        self.pipe = StableDiffusionPipeline.from_single_file("https://huggingface.co/Onlyfake/realistic-vision-v4/blob/main/Realistic_Vision_V4.0.safetensors")
+        self.pipe = self.pipe.to(device)
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]: