Spaces:

seddiktrk
/

CLIP-GPT_Image_Captionning

Sleeping

seddiktrk commited on Aug 31, 2024

Commit

ce151ef

verified ·

1 Parent(s): 4a09c08

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -172,7 +172,7 @@ path = "model_epoch_1.pt"
 state_dict = torch.load(path)
 # Apply the weights to the model
-CliPGPT.load_state_dict(state_dict)
 CliPGPT.to(device)
 print('importing CLIP')

 state_dict = torch.load(path)
 # Apply the weights to the model
+CliPGPT.load_state_dict(state_dict,map_location=torch.device('cpu') )
 CliPGPT.to(device)
 print('importing CLIP')