Upload folder using huggingface_hub
Browse files
train.py
CHANGED
|
@@ -43,11 +43,11 @@ tokenizer = AutoTokenizer.from_pretrained(
|
|
| 43 |
)
|
| 44 |
tokenizer.pad_token = tokenizer.eos_token
|
| 45 |
|
| 46 |
-
def formatting_func(example):
|
| 47 |
-
|
| 48 |
-
|
| 49 |
-
def generate_and_tokenize_prompt(prompt):
|
| 50 |
-
|
| 51 |
|
| 52 |
|
| 53 |
bnb_config = BitsAndBytesConfig(
|
|
|
|
| 43 |
)
|
| 44 |
tokenizer.pad_token = tokenizer.eos_token
|
| 45 |
|
| 46 |
+
# def formatting_func(example):
|
| 47 |
+
# text = f"### The following is a note by Eevee the Dog: {example['note']}"
|
| 48 |
+
# return text
|
| 49 |
+
# def generate_and_tokenize_prompt(prompt):
|
| 50 |
+
# return tokenizer(formatting_func(prompt))
|
| 51 |
|
| 52 |
|
| 53 |
bnb_config = BitsAndBytesConfig(
|