Alexmikupro
/

vtuber-maker

StableDiffusionPipeline

Model card Files Files and versions

Alexmikupro commited on Oct 31, 2023

Commit

f041201

·

1 Parent(s): 6b539a6

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -36,7 +36,7 @@ class EndpointHandler():
         self.model.eval()
         # Create a RealESRGANer object for inference
-        self.upsampler = RealESRGANer(scale=4, model=self.model, tile=0, model_path=esrgan_model_path)
     def __call__(self, data: Dict[str, Any], output_size=(512, )) -> Dict[str, str]:

         self.model.eval()
         # Create a RealESRGANer object for inference
+        self.upsampler = RealESRGANer(scale=4, model=self.model, tile=0, model_path="/repository/RealESRGAN_x4plus_anime_6B.pth")
     def __call__(self, data: Dict[str, Any], output_size=(512, )) -> Dict[str, str]: