Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -40,10 +40,10 @@ def group_texts(examples):
|
|
40 |
#Access-Token (in Secrets)
|
41 |
|
42 |
#access_token = st.secrets["ACCESS_TOKEN"]
|
43 |
-
access_token = "
|
44 |
login(token="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")
|
45 |
#Modelle und Tokenizer
|
46 |
-
|
47 |
#Alternativ mit beliebigen Modellen:
|
48 |
base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
|
49 |
#base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
|
@@ -88,7 +88,7 @@ lm_datasets = tokenized_datasets.map(
|
|
88 |
#Training Args
|
89 |
model_name = base_model.split("/")[-1]
|
90 |
training_args = TrainingArguments(
|
91 |
-
"/model",
|
92 |
evaluation_strategy = "epoch",
|
93 |
learning_rate=2e-5,
|
94 |
weight_decay=0.01,
|
|
|
40 |
#Access-Token (in Secrets)
|
41 |
|
42 |
#access_token = st.secrets["ACCESS_TOKEN"]
|
43 |
+
access_token = "hf_..."
|
44 |
login(token="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")
|
45 |
#Modelle und Tokenizer
|
46 |
+
model_name = "alexkueck/LIFineTuned"
|
47 |
#Alternativ mit beliebigen Modellen:
|
48 |
base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_and_model)
|
49 |
#base_model = "TheBloke/airoboros-13B-HF" #load_8bit = False (in load_tokenizer_and_model)
|
|
|
88 |
#Training Args
|
89 |
model_name = base_model.split("/")[-1]
|
90 |
training_args = TrainingArguments(
|
91 |
+
f"{model_name}/model",
|
92 |
evaluation_strategy = "epoch",
|
93 |
learning_rate=2e-5,
|
94 |
weight_decay=0.01,
|