Getting this issue... TrainingConfig can't load

[AMEERAZAM08]

`ImportError Traceback (most recent call last)
in <cell line: 2>()
1 import torch
----> 2 from mario_gpt import MarioDataset, MarioLM, TrainingConfig, MarioGPTTrainer
3
4 # create basic gpt model
5 BASE = "distilgpt2"

ImportError: cannot import name 'TrainingConfig' from 'mario_gpt' (/usr/local/lib/python3.10/dist-packages/mario_gpt/init.py)`

[AMEERAZAM08]

Please if possible @shyamsn97 help me in this I'm training

[AMEERAZAM08]

Any update @shyamsn97 ?

[AMEERAZAM08]

Issue resolve but getting this now ..

TypeError Traceback (most recent call last)
Cell In[11], line 1
----> 1 trainer = MarioGPTTrainer(mario_lm, dataset, config=config)

File c:\ProgramData\miniconda3\envs\mario-gpt\lib\site-packages\mario_gpt-0.1.3-py3.8.egg\mario_gpt\trainer.py:73, in MarioGPTTrainer.init(self, mario_lm, train_dataset, config, optimizer, lr_scheduler)
70 if lr_scheduler is None:
71 self.lr_scheduler = self.create_lr_scheduler(self.config, self.optimizer)
---> 73 self.accelerator = self.create_accelerator(self.config)

File c:\ProgramData\miniconda3\envs\mario-gpt\lib\site-packages\mario_gpt-0.1.3-py3.8.egg\mario_gpt\trainer.py:92, in MarioGPTTrainer.create_accelerator(self, config)
91 def create_accelerator(self, config: Any) -> Accelerator:
---> 92 return Accelerator(
93 mixed_precision=config.mixed_precision,
94 gradient_accumulation_steps=config.gradient_accumulation_steps,
95 log_with="tensorboard",
96 logging_dir=os.path.join(config.output_dir, "logs"),
97 )

TypeError: init() got an unexpected keyword argument 'logging_dir'

[AMEERAZAM08]

sorted

[AMEERAZAM08]

use accelerate 9.3.0 version