cpu map location

app.py

@@ -121,7 +121,7 @@ def generate_default_card(img_size=(300, 300)):
 # checkpoint = torch.load('/Users/dylan.d/Documents/research/genes/zinc20++/ckpt.pt')
 
 tokenizer = PreTrainedTokenizerFast(tokenizer_file='tokenizer_4096.json')
-checkpoint = torch.load('ckpt.pt')
+checkpoint = torch.load('ckpt.pt', map_location=torch.device('cpu'))
 
 args = ModelArgs(**checkpoint['model_args'])
 model = Transformer(args)