Spaces:
Sleeping
Sleeping
Upload folder using huggingface_hub
Browse files
train.py
CHANGED
|
@@ -8,6 +8,8 @@ from peft import LoraConfig, get_peft_model, prepare_model_for_kbit_training
|
|
| 8 |
from trl import SFTTrainer
|
| 9 |
|
| 10 |
def train():
|
|
|
|
|
|
|
| 11 |
hf_token = os.environ.get("HF_TOKEN")
|
| 12 |
if not hf_token:
|
| 13 |
print("Empty HF_TOKEN. Aborting.")
|
|
|
|
| 8 |
from trl import SFTTrainer
|
| 9 |
|
| 10 |
def train():
|
| 11 |
+
global torch
|
| 12 |
+
print("Code Version: 1.0.2 (Shadow Fix)", flush=True)
|
| 13 |
hf_token = os.environ.get("HF_TOKEN")
|
| 14 |
if not hf_token:
|
| 15 |
print("Empty HF_TOKEN. Aborting.")
|