Kevin Fink
commited on
Commit
·
e4474ca
1
Parent(s):
8786178
deve
Browse files
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 |
)
|