diff --git a/imaginairy/cli/shared.py b/imaginairy/cli/shared.py
index 0ccd55a..220ddb4 100644
--- a/imaginairy/cli/shared.py
+++ b/imaginairy/cli/shared.py
@@ -23,6 +23,8 @@ def imaginairy_click_context(log_level="INFO"):
         yield
     except errors_to_catch as e:
         logger.error(e)
+        # import traceback
+        # traceback.print_exc()
 
 
 def _imagine_cmd(
diff --git a/imaginairy/utils/__init__.py b/imaginairy/utils/__init__.py
index cd4146d..a41cb81 100644
--- a/imaginairy/utils/__init__.py
+++ b/imaginairy/utils/__init__.py
@@ -1,6 +1,5 @@
 import importlib
 import logging
-import numpy as np
 import platform
 import random
 import re
@@ -9,6 +8,7 @@ from contextlib import contextmanager, nullcontext
 from functools import lru_cache
 from typing import Any, List, Optional
 
+import numpy as np
 import torch
 from torch import Tensor, autocast
 from torch.nn import functional
@@ -337,6 +337,7 @@ def clear_gpu_cache():
     if torch.cuda.is_available():
         torch.cuda.empty_cache()
 
+
 def seed_everything(seed: int | None = None) -> None:
     if seed is None:
         seed = random.randint(0, 2**32 - 1)
@@ -344,4 +345,4 @@ def seed_everything(seed: int | None = None) -> None:
     random.seed(a=seed)
     np.random.seed(seed=seed)
     torch.manual_seed(seed=seed)
-    torch.cuda.manual_seed_all(seed=seed)
\ No newline at end of file
+    torch.cuda.manual_seed_all(seed=seed)
diff --git a/imaginairy/weight_management/conversion.py b/imaginairy/weight_management/conversion.py
index b26e319..1467068 100644
--- a/imaginairy/weight_management/conversion.py
+++ b/imaginairy/weight_management/conversion.py
@@ -74,7 +74,10 @@ class WeightMap:
     def cast_weights(self, source_weights) -> dict[str, "Tensor"]:
         converted_state_dict: dict[str, "Tensor"] = {}
         for source_key in source_weights:
-            source_prefix, suffix = source_key.rsplit(sep=".", maxsplit=1)
+            try:
+                source_prefix, suffix = source_key.rsplit(sep=".", maxsplit=1)
+            except ValueError:
+                continue
             # handle aliases
             source_prefix = self.source_aliases.get(source_prefix, source_prefix)
             try: