hssling commited on
Commit
b43cd1f
·
verified ·
1 Parent(s): 7b299f1

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. train.py +2 -0
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.")