Spaces:

AlexWortega
/

food_calories

Build error

AlexWortega commited on Jan 12, 2022

Commit

0127e28

1 Parent(s): 751b158

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from rudolph import utils
 device = 'cpu'
 model = get_rudolph_model('350M',  fp16=True, device=device)
-model.load_state_dict(torch.load("awesomemodel__dalle_1500.pt"))
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=False).to(device)

 device = 'cpu'
 model = get_rudolph_model('350M',  fp16=True, device=device)
+model.load_state_dict(torch.load("awesomemodel__dalle_1500.pt",map_location=torch.device('cpu')))
 tokenizer = get_tokenizer()
 vae = get_vae(dwt=False).to(device)