bump versions

requirements.txt

@@ -27,13 +27,13 @@ nvitop==1.3.2
 # latest versions
 tensorboard
 peft==0.10.0
-trl==0.8.1
+trl==0.8.5
 tiktoken==0.6.0
-transformers==4.39.3
+transformers==4.40.0
 
-accelerate==0.29.1
+accelerate==0.29.3
 diffusers==0.27.2
-bitsandbytes==0.43.0
+bitsandbytes==0.43.1
 # extras
 rouge_score==0.1.2
 py7zr==0.20.6
@@ -47,3 +47,4 @@ pyngrok==7.0.3
 authlib==1.3.0
 itsdangerous==2.1.2
 seqeval==1.2.2
+

src/autotrain/__init__.py

@@ -41,4 +41,4 @@
 
 
 logger = Logger().get_logger()
-__version__ = "0.7.53.dev0"
+__version__ = "0.7.54.dev0"

src/autotrain/preprocessor/text.py

@@ -80,6 +80,9 @@ def prepare(self):
         train_df, valid_df = self.split()
         train_df, valid_df = self.prepare_columns(train_df, valid_df)
 
+        train_df[:, "autotrain_label"] = train_df["autotrain_label"].astype(str)
+        valid_df[:, "autotrain_label"] = valid_df["autotrain_label"].astype(str)
+
         label_names = sorted(set(train_df["autotrain_label"].unique().tolist()))
 
         train_df = Dataset.from_pandas(train_df)