fix

generator.py

@@ -164,26 +164,8 @@ class Generator:
 
 
 def load_csm_1b(device: str = "cuda") -> Generator:
-    try:
-        # Try the simple approach first
-        model = Model.from_pretrained("sesame/csm-1b")
-        model = model.to(device=device)
-        
-        generator = Generator(model)
-        return generator
-    except Exception as e:
-        # Log the error for debugging
-        import logging
-        logging.error(f"Error in standard model loading: {str(e)}")
-        
-        # Try alternative loading if config is available
-        try:
-            from silentcipher import Config
-            model_path = "sesame/csm-1b"
-            config = Config.from_pretrained(model_path)
-            model = Model.from_pretrained(model_path, config=config)
-            model = model.to(device=device)
-            return Generator(model)
-        except ImportError:
-            # Config not available, try direct initialization
-            raise RuntimeError("Could not load model with either method. Original error: " + str(e))
+    model = Model.from_pretrained("sesame/csm-1b")
+    model.to(device=device, dtype=torch.bfloat16)
+
+    generator = Generator(model)
+    return generator