alexkueck commited on
Commit
864967c
·
1 Parent(s): c91b344

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -22,6 +22,8 @@ base_model = "project-baize/baize-v2-7b" #load_8bit = False (in load_tokenizer_
22
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
23
  tokenizer,model,device = load_tokenizer_and_model(base_model,False)
24
 
 
 
25
 
26
  ########################################################################
27
  #Chat KI nutzen, um Text zu generieren...
 
22
  #base_model = "TheBloke/airoboros-13B-HF" #load_8bit = True
23
  tokenizer,model,device = load_tokenizer_and_model(base_model,False)
24
 
25
+ tokenized_datasets = datasets.map(tokenize_function, batched=True, num_proc=4, remove_columns=["text"])
26
+
27
 
28
  ########################################################################
29
  #Chat KI nutzen, um Text zu generieren...