SpicyMelonYT commited on
Commit
846bac8
·
1 Parent(s): 0925243

string the token

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -44,15 +44,16 @@ def respond(
44
 
45
 
46
  def train_model():
47
-
48
- os.environ["HUGGINGFACE_TOKEN"] = hf_token
49
 
50
  # Load dataset
51
  dataset = load_dataset('json', data_files={
52
  'train': 'training_set.json'})
53
 
54
  # Load model
55
- model = AutoModelForCausalLM.from_pretrained('meta-llama/Meta-Llama-3-8B-Instruct')
 
56
 
57
  # Define training arguments
58
  training_args = TrainingArguments(
 
44
 
45
 
46
  def train_model():
47
+ print(str(hf_token), flush=True)
48
+ os.environ["HUGGINGFACE_TOKEN"] = str(hf_token)
49
 
50
  # Load dataset
51
  dataset = load_dataset('json', data_files={
52
  'train': 'training_set.json'})
53
 
54
  # Load model
55
+ model = AutoModelForCausalLM.from_pretrained(
56
+ 'meta-llama/Meta-Llama-3-8B-Instruct')
57
 
58
  # Define training arguments
59
  training_args = TrainingArguments(