ameerazam08 commited on
Commit
688daa5
1 Parent(s): ed21a70

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. train.py +5 -5
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
- 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(
 
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(