Spaces:

medmac01
/

stable-diff-multilingual-v0.1

Sleeping

medmac01 commited on Mar 7, 2024

Commit

0b16ac5

verified ·

1 Parent(s): 1f7998d

map weights to cpu

Files changed (1) hide show

model.py CHANGED Viewed

@@ -76,7 +76,7 @@ class AdaptationLayer(nn.Module):
 adapt_model  = AdaptationLayer(768,768)
 adapt_model.to(device)
-state_dict = torch.load('weights/checkpoint_9.pth')
 adapt_model.load_state_dict(state_dict)
 from Multilingual_CLIP.multilingual_clip import pt_multilingual_clip

 adapt_model  = AdaptationLayer(768,768)
 adapt_model.to(device)
+state_dict = torch.load('weights/checkpoint_9.pth', map_location=torch.device('cpu'))
 adapt_model.load_state_dict(state_dict)
 from Multilingual_CLIP.multilingual_clip import pt_multilingual_clip