Merge pull request #66 from Ivan-Inby/bugfix/fallback-to-cpu-if-no-cuda

fix: fallback to CPU when CUDA is not available

chat.py

@@ -93,7 +93,8 @@ class DOLPHIN:
             ckpt = try_rename_lagacy_weights(ckpt)
             self.model.load_state_dict(ckpt, strict=True)
 
-        self.model.to("cuda")
+        device = "cuda" if torch.cuda.is_available() else "cpu"
+        self.model.to(device)
         self.model.eval()
         transform_args = {
             "input_size": self.swin_args["img_size"],