dev

app.py

@@ -110,9 +110,9 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
                 #text_target=examples['target'],
                 return_tensors='pt', 
             )
-            labels["input_ids"] = [
-                [(l if l != tokenizer.pad_token_id else -100) for l in label] for label in labels["input_ids"]
-            ]        
+            #labels["input_ids"] = [
+             #   [(l if l != tokenizer.pad_token_id else -100) for l in label] for label in labels["input_ids"]
+            #]        
             # Add labels to the model inputs
             model_inputs["labels"] = labels["input_ids"]
             return model_inputs