burtenshaw HF Staff commited on
Commit
72bcd86
·
verified ·
1 Parent(s): eaead4f

Upload train.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. train.py +19 -0
train.py CHANGED
@@ -1,3 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  """## Import libraries"""
2
 
3
  import torch
 
1
+ # /// script
2
+ # requires-python = ">=3.10"
3
+ # dependencies = [
4
+ # "datasets",
5
+ # "httpx",
6
+ # "huggingface - hub",
7
+ # "setuptools",
8
+ # "transformers",
9
+ # "torch",
10
+ # "accelerate",
11
+ # "trl",
12
+ # "peft",
13
+ # "wandb",
14
+ # "torchvision",
15
+ # "torchaudio"
16
+ # ]
17
+ # ///
18
+
19
+
20
  """## Import libraries"""
21
 
22
  import torch