sunkencity commited on
Commit
9b26fb6
·
verified ·
1 Parent(s): fe50797

Upload train_aviation.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train_aviation.py +3 -2
train_aviation.py CHANGED
@@ -1,9 +1,10 @@
1
  # /// script
2
  # dependencies = [
3
- # "torch", # Ensure torch is installed before flash-attn
4
  # "trl>=0.12.0",
5
  # "peft>=0.7.0",
6
- # "transformers>=4.36.0",
 
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",