NameError: name 'OctoInference' is not defined.

[armstronglzh]

I have download the octo files according to Octo Inference Setup,why still no octo model found

[armstronglzh]

by the way I am running command:
python simpler_env/main_inference.py --policy-model octo-small --ckpt-path None --robot widowx --policy-setup widowx_bridge --control-freq 5 --sim-freq 500 --max-episode-steps 60 --env-name StackGreenCubeOnYellowCubeBakedTexInScene-v0 --scene-name bridge_table_1_v1 --rgb-overlay-path ManiSkill2_real2sim/data/real_inpainting/bridge_real_eval_1.png --robot-init-x 0.147 0.147 1 --robot-init-y 0.028 0.028 1 --obj-variation-mode episode --obj-episode-range 0 24 --robot-init-rot-quat-center 0 0 0 1 --robot-init-rot-rpy-range 0 0 1 0 0 1 0 0 1 --additional-env-save-tags octo_init_rng_0

[armstronglzh]

And the folder is like:
SimplerEnv-main-octo-docs
SimplerEnv-main-octo-examples
SimplerEnv-main-simpler_env-main_inference.py
SimplerEnv-main-octo-octo-model-octo_model.py,this is the py file quoted by the "octo_model" py file which is quoted by the main_inference.py:
try:
from simpler_env.policies.octo.octo_model import OctoInference
except ImportError as e:
print("Octo is not correctly imported.")
print(e)

[xuanlinli17]

I think you forgot to cd octo/; pip install -e .

Please follow the instruciton in the Octo inference setup in README (it also requires installing jax)

[armstronglzh]

I think you forgot to cd octo/; pip install -e .

Please follow the instruciton in the Octo inference setup in README (it also requires installing jax)

I am sure I have done this all,the "pip install -e ." result is:
Successfully built octo
Installing collected packages: octo
Successfully installed octo-0.0.0

but still the error