Spaces:
Sleeping
Sleeping
SpicyMelonYT
commited on
Commit
·
846bac8
1
Parent(s):
0925243
string the token
Browse files
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(
|
|
|
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(
|