Add dynamic dtype selection for `get_device_info()` with CUDA/CPU checks, update config logic

src/config.py

@@ -82,7 +82,7 @@ def get_device_info() -> tuple[str, str]:
 
 
 device, device_description = get_device_info()
-dtype = torch.float16
+dtype = torch.float16 if device == "cuda" else torch.float32
 
 # Check if we're on low-memory hardware and warn
 def check_memory_requirements() -> bool: