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...
|