Kevin Fink commited on
Commit
e4474ca
·
1 Parent(s): 8786178
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -109,7 +109,7 @@ def fine_tune_model(model, dataset_name, hub_id, api_key, num_epochs, batch_size
109
  examples['text'],
110
  #max_length=max_length, # Set to None for dynamic padding
111
  truncation=True,
112
- padding='max_length',
113
  #return_tensors='pt',
114
  padding=True,
115
  )
 
109
  examples['text'],
110
  #max_length=max_length, # Set to None for dynamic padding
111
  truncation=True,
112
+ #padding='max_length',
113
  #return_tensors='pt',
114
  padding=True,
115
  )