search

abdulhaim / LMRL-Gym

MIT License
71 stars 9 forks source link

jax_models problems #2

Closed DT6A closed 7 months ago

DT6A commented 9 months ago

Hello! I'm trying to run the llm_rl_scripts/car_dealer/bc/train_bc.py but it fails with the following error:

Traceback (most recent call last):
  File "/home/user/LMRL-Gym/llm_rl_scripts/car_dealer/bc/train_bc.py", line 6, in <module>
    from jax_models.gpt2 import load_gpt2_model
ModuleNotFoundError: No module named 'jax_models'

I followed all of the installation instructions. Any suggestions what could go wrong?

icwhite commented 9 months ago

Hi DT6A, thank you for letting us know about the issue! Could you let us know what hardware you are running on? We will try to reproduce the issue.

My guess is that the wrong version of a package is installed. The installation instructions are different depending on whether you are running on CPU, GPU or TPU.

DT6A commented 9 months ago

Hi, I was trying CPU installation

DT6A commented 9 months ago

@icwhite hi! Any updates on that?

icwhite commented 9 months ago

Hi! This is an issue with an older version of the package. We will fix this problem soon. Thank you for your patience.

DT6A commented 9 months ago

@icwhite, OK, thanks for your response!

icwhite commented 7 months ago

Hi @DT6A! Apologies for the delay but the bc script for Car Dealer is now functional!

DT6A commented 7 months ago

Thank you!