alexkueck commited on
Commit
a539a3b
·
1 Parent(s): 424c3fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = "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"
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,