alexkueck commited on
Commit
c50213d
·
1 Parent(s): 31cca09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -18,7 +18,9 @@ base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_
18
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
19
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
20
  tokenizer,model,device = load_tokenizer_and_model(base_model,False)
21
- dataset_neu = daten_laden("alexkueck/tis")
 
 
22
 
23
  #############################################
24
  #Vorbereiten für das Training der neuen Daten
 
18
  #base_model = "EleutherAI/gpt-neo-1.3B" #load_8bit = False (in load_tokenizer_and_model)
19
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
20
  tokenizer,model,device = load_tokenizer_and_model(base_model,False)
21
+
22
+ #dataset_neu = daten_laden("alexkueck/tis")
23
+ dataset_neu = daten_laden("EleutherAI/pile")
24
 
25
  #############################################
26
  #Vorbereiten für das Training der neuen Daten