Alexmikupro
/

vtuber-maker

StableDiffusionPipeline

Model card Files Files and versions

Alexmikupro commited on Oct 31, 2023

Commit

6b539a6

·

1 Parent(s): f3842fa

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -21,7 +21,7 @@ class EndpointHandler():
         self.pipe = self.pipe.to(device)
         # Load the ESRGAN state dictionary
-        checkpoint = torch.load("Alexmikupro/vtuber-maker/RealESRGAN_x4plus_anime_6B.pth")
         # Check if 'params_ema' is in the keys and filter the state_dict
         if "params_ema" in checkpoint:

         self.pipe = self.pipe.to(device)
         # Load the ESRGAN state dictionary
+        checkpoint = torch.load("/repository/RealESRGAN_x4plus_anime_6B.pth")
         # Check if 'params_ema' is in the keys and filter the state_dict
         if "params_ema" in checkpoint: