mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 09:46:00 +00:00
remove test
This commit is contained in:
@@ -1,27 +0,0 @@
|
|||||||
#!/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)
|
|
||||||
Reference in New Issue
Block a user