Spaces:
Paused
Paused
Update app.py
Browse files
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...
|