replicate/cog

cog.server.exceptions.FatalWorkerException: Predictor errored during setup: Ran out of input

Closed this issue · 0 comments

Hi @mattt , @zeke ..


class Predictor(BasePredictor):

    def setup(self):

        os.makedirs('weights', exist_ok=True)
        os.chdir('weights')
        if not os.path.exists('Swin2SR_ClassicalSR_X4_64.pth'):
            os.system(
                'wget https://github.com/mv-lab/swin2sr/releases/download/v0.0.1/Swin2SR_ClassicalSR_X4_64.pth'
            )
        if not os.path.exists('GFPGANv1.4.pth'):
            os.system(
                'wget https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth'
            )
        if not os.path.exists('inswapper_128.onnx'):
            os.system(
                'wget https://huggingface.co/ashleykleynhans/inswapper/resolve/main/inswapper_128.onnx'
            )
        if not os.path.exists('Swin2SR_CompressedSR_X4_48.pth'):
            os.system(
                'wget https://github.com/mv-lab/swin2sr/releases/download/v0.0.1/Swin2SR_CompressedSR_X4_48.pth'
            )
        if not os.path.exists('Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR.pth'):
            os.system(
                'wget https://github.com/mv-lab/swin2sr/releases/download/v0.0.1/Swin2SR_RealworldSR_X4_64_BSRGAN_PSNR.pth'
            )

        os.chdir('..')
        if not os.path.exists('model.pth'):
            os.system(
                'wget https://github.com/codeplugtech/Background_remover/releases/download/v0.0.1/model.pth'
            )

i have load multi modles in cog it works finely in locall...when i pushed into an replicate below setup error occures..
![Screenshot (63)](https://github.com/replicate/cog/assets/99823984/88741f13-af84-48c7-be58-bdb17fee40d3)