ValueError: No configurable matching 'transformer_layer.TransformerLayerGenerate'. In file "geometry_150M_generate.gin", line 11 transformer_layer.TransformerLayerGenerate

[Janam-netizen]

I am unable to run the lm_infererence_test.py script . I have installed the meliad library adnd runnnng the file as mentioned in the run_tests.sh bash script. I have intalled the meliad library and ag_ckpt_vocab as data. I am using macos and python 3.10.14 and have installes all the libraries in requirenment.txt withou any issues

  jax.tree_util.register_keypaths(data_clz, keypaths)
Running tests under Python 3.10.14: /Users/janamzaveri/KRR/alphageometry/krr_proj/bin/python
I0807 03:12:50.226891 8626654208 inference_utils.py:69] Parsing gin configuration.
I0807 03:12:50.226947 8626654208 inference_utils.py:71] Added Gin search path meliad_lib/meliad/transformer/configs
I0807 03:12:50.227056 8626654208 inference_utils.py:71] Added Gin search path /Users/janamzaveri/KRR/alphageometry
I0807 03:12:50.227075 8626654208 inference_utils.py:74] Loading Gin config file base_htrans.gin
I0807 03:12:50.227090 8626654208 inference_utils.py:74] Loading Gin config file size/medium_150M.gin
I0807 03:12:50.227107 8626654208 inference_utils.py:74] Loading Gin config file options/positions_t5.gin
I0807 03:12:50.227123 8626654208 inference_utils.py:74] Loading Gin config file options/lr_cosine_decay.gin
I0807 03:12:50.227138 8626654208 inference_utils.py:74] Loading Gin config file options/seq_1024_nocache.gin
I0807 03:12:50.227151 8626654208 inference_utils.py:74] Loading Gin config file geometry_150M_generate.gin
I0807 03:12:50.227166 8626654208 inference_utils.py:76] Overriding Gin param DecoderOnlyLanguageModelGenerate.output_token_losses=True
I0807 03:12:50.227182 8626654208 inference_utils.py:76] Overriding Gin param TransformerTaskConfig.batch_size=2
I0807 03:12:50.227196 8626654208 inference_utils.py:76] Overriding Gin param TransformerTaskConfig.sequence_length=128
I0807 03:12:50.227211 8626654208 inference_utils.py:76] Overriding Gin param Trainer.restore_state_variables=False
[  FAILED  ] setUpClass (__main__.LmInferenceTest)
======================================================================
ERROR: setUpClass (__main__.LmInferenceTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/Users/janamzaveri/KRR/alphageometry/lm_inference_test.py", line 59, in setUpClass
    lm.parse_gin_configuration(gin_file, gin_param, gin_paths=gin_search_paths)
  File "/Users/janamzaveri/KRR/alphageometry/meliad_lib/meliad/transformer/inference_utils.py", line 77, in parse_gin_configuration
    gin.parse_config_files_and_bindings(gin_files, gin_params)
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/config.py", line 2497, in parse_config_files_and_bindings
    includes_and_imports = parse_config_file(config_file, skip_unknown)
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/config.py", line 2450, in parse_config_file
    includes, imports = parse_config(f, skip_unknown=skip_unknown)
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/config.py", line 2334, in parse_config
    with utils.try_with_location(statement.location):
  File "/opt/homebrew/Cellar/[email protected]/3.10.14_1/Frameworks/Python.framework/Versions/3.10/lib/python3.10/contextlib.py", line 153, in __exit__
    self.gen.throw(typ, value, traceback)
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/utils.py", line 60, in try_with_location
    augment_exception_message_and_reraise(exception, _format_location(location))
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/utils.py", line 41, in augment_exception_message_and_reraise
    raise proxy.with_traceback(exception.__traceback__) from None
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/utils.py", line 56, in try_with_location
    yield
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/config.py", line 2336, in parse_config
    _raise_unknown_configurable_error(statement.selector)
  File "/Users/janamzaveri/KRR/alphageometry/krr_proj/lib/python3.10/site-packages/gin/config.py", line 648, in _raise_unknown_configurable_error
    raise ValueError(f"No configurable matching '{selector}'.")
ValueError: No configurable matching 'transformer_layer.TransformerLayerGenerate'.
  In file "geometry_150M_generate.gin", line 11
    transformer_layer.TransformerLayerGenerate:

----------------------------------------------------------------------
Ran 0 tests in 0.007s