English

Model Loading Issue.

#3
by pk80103 - opened

image
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

Sign up or log in to comment