Upload train.py with huggingface_hub
Browse files
    	
        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
         | 
