map location of torch load to cpu

server.py

@@ -71,7 +71,7 @@ model = Dalle.from_pretrained("minDALL-E/1.3B")  # This will automatically downl
 # model.stage2.load_state_dict(dalle_stage_dict)
 
 # NEW METHOD
-model.load_state_dict(torch.load(full_dict_path))
+model.load_state_dict(torch.load(full_dict_path, map_location=torch.device('cpu')))
 model.to(device=device)
 
 # model_clip, preprocess_clip = clip.load("ViT-B/32", device=device)