search

dusty-nv / jetson-containers

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T
MIT License
2.36k stars 481 forks source link

There is no preset of `OpenVLA-MimicGen-FP16` #602

Closed junh0cho closed 2 months ago

junh0cho commented 2 months ago

I am following tutorial of https://www.jetson-ai-lab.com/openvla.html

jetson-containers run $(autotag nano_llm) \
  python3 -m nano_llm.studio --load OpenVLA-MimicGen-FP16

Here OpenVLA-MimicGen-FP16 is not available , along with OpenVLA-MimicGen-FP8, OpenVLA-MimicGen-INT4. They seem to be available in https://github.com/dusty-nv/jetson-containers/tree/master/data/nano_llm/presets

dusty-nv commented 2 months ago

Ahh thank you @junh0cho, I had forgotten to merge these from the dev branch - here they are now in jetson-containers master: https://github.com/dusty-nv/jetson-containers/commit/0d9e9a4d7f70841e03ed6fa9e71869bd7888b78b

junh0cho commented 2 months ago

Thanks! it works!