Model Loading Issue.
#3
by
pk80103
- opened

Hi Team,
I am trying to load the model but it fails without any error message, I understand it can be because of limited RAM or GPU.
How much RAM and which GPU is required to test the small model.
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = SAMAudio.from_pretrained("facebook/sam-audio-small").to(device).eval()
Thanks
Praveen