add pi05

test_pi05_hub_only.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python
+# TODO(pepijn): delete this file
+"""Quick test script to load and test only the PI0.5 model from HuggingFace hub."""
+
+from test_pi0_hub import test_hub_loading
+
+
+def main():
+    """Test only the PI0.5 model."""
+    print("\n")
+    print("=" * 60)
+    print("PI0.5 Model Quick Test")
+    print("=" * 60)
+
+    success = test_hub_loading(model_id="pepijn223/pi05_base_fp32", model_name="PI0.5")
+
+    if success:
+        print("\n✅ PI0.5 model loaded and tested successfully!")
+    else:
+        print("\n❌ PI0.5 test failed!")
+
+    return success
+
+
+if __name__ == "__main__":
+    success = main()
+    exit(0 if success else 1)