Upload test_model.py

test_model.py

@@ -0,0 +1,21 @@
+"""Quick sanity test of LuminaRS without heavy deps."""
+import torch
+from luminars.model import LuminaRS
+from luminars.config import LuminaRSConfig
+
+def test():
+    cfg = LuminaRSConfig()
+    model = LuminaRS(cfg)
+    n = sum(p.numel() for p in model.parameters())
+    print(f"Total params: {n/1e6:.1f}M")
+    print(f"Trainable params: {sum(p.numel() for p in model.parameters() if p.requires_grad)/1e6:.1f}M")
+
+    bs, d, h, w = 2, cfg.latent_dim, cfg.latent_h, cfg.latent_w
+    z = torch.randn(bs, d, h, w)
+    text = torch.randn(bs, cfg.max_text_len, cfg.text_embed_dim)
+    t = torch.rand(bs)
+    out = model(z, text, t)
+    print(f"Forward OK, output shape: {out.shape}")
+
+if __name__ == "__main__":
+    test()