New TypeError: __init__() got an unexpected keyword argument 'progress_bar_refresh_rate'

[aishwaryapisal9]

I have imported the model and necessary libraries. I am getting the below error in google colab. I have used this model earlier also few months back and it was working fine. This is the new issue I am facing recently with the same code.

---

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

Imported libraries:

!pip install keytotext --upgrade
!sudo apt-get install git-lfs

from keytotext import trainer

Training Model:

model = trainer()
model.from_pretrained(model_name="t5-small")
model.train(train_df=df_train_final, test_df=df_test, batch_size=3, max_epochs=5,use_gpu=True)
model.save_model()

Have attached error screenshot

• OS: Windows
• Browser Chrome
  

[saied71]

I did a workaround for this problem.
I first installed ket2text as you did then clone the main branch and deleted progress_bar_refresh_rate as mentioned here. you can see the fork I used here

after that, I import trainer directly from the module. In other words, I created a trainer.py at the root of my notebook and imported it.
So the code is :

import pandas as pd
from trainer import trainer
from transformers import  AutoModelForSeq2SeqLM

data = pd.read_csv("samples.csv")
df = data.copy()
df.rename({"sents":"text", "kws":"keywords"}, axis=1, inplace=True)

train_df = df[:200]
test_df = df[200:250]


model = trainer()
model.from_pretrained(model_name="t5-small")

model.train(train_df=train_df, test_df=test_df,use_gpu=True,batch_size=1, max_epochs=3)

[Mukish45]

I did a workaround for this problem. I first installed ket2text as you did then clone the main branch and deleted progress_bar_refresh_rate as mentioned here. you can see the fork I used here

after that, I import trainer directly from the module. In other words, I created a trainer.py at the root of my notebook and imported it. So the code is :

import pandas as pd
from trainer import trainer
from transformers import  AutoModelForSeq2SeqLM

data = pd.read_csv("samples.csv")
df = data.copy()
df.rename({"sents":"text", "kws":"keywords"}, axis=1, inplace=True)

train_df = df[:200]
test_df = df[200:250]


model = trainer()
model.from_pretrained(model_name="t5-small")

model.train(train_df=train_df, test_df=test_df,use_gpu=True,batch_size=1, max_epochs=3)

I am facing two issues.
One, I can't use the GPU to run this program.
Second, I'm facing issue in uploading the model. Can you able to upload and load the model?