Upload train_aviation.py with huggingface_hub
Browse files- train_aviation.py +3 -2
train_aviation.py
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
# /// script
|
| 2 |
# dependencies = [
|
| 3 |
-
# "torch",
|
| 4 |
# "trl>=0.12.0",
|
| 5 |
# "peft>=0.7.0",
|
| 6 |
-
# "transformers>=4.
|
|
|
|
| 7 |
# "accelerate>=0.24.0",
|
| 8 |
# "trackio",
|
| 9 |
# "bitsandbytes",
|
|
|
|
| 1 |
# /// script
|
| 2 |
# dependencies = [
|
| 3 |
+
# "torch",
|
| 4 |
# "trl>=0.12.0",
|
| 5 |
# "peft>=0.7.0",
|
| 6 |
+
# "transformers>=4.46.0",
|
| 7 |
+
# "huggingface_hub>=0.26.0",
|
| 8 |
# "accelerate>=0.24.0",
|
| 9 |
# "trackio",
|
| 10 |
# "bitsandbytes",
|