Audio-AGI / AudioSep

Official implementation of "Separate Anything You Describe"
https://audio-agi.github.io/Separate-Anything-You-Describe/
MIT License
1.63k stars 118 forks source link

Problem with the pipiline #42

Closed Brodvd closed 1 month ago

Brodvd commented 2 months ago

Hi @liuxubo717 I ranpipiline.py but where I find the checkpoint 3920000.ckpt? With the others checkpoint make error.


if __name__ == '__main__':
    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    model = build_audiosep(
        config_yaml='config/audiosep_base.yaml', 
        checkpoint_path='checkpoint/step=3920000.ckpt', 
        device=device)