github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-06-24 06:27:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #268 from MalikMAlna/dev

Browse Source 
Slight cleanup
This commit is contained in:
Andriy Mulyar 2023-04-07 10:50:56 -04:00 committed by GitHub
commit 8e28a33731
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
data.py
View File

@ -31,7 +31,7 @@ def tokenize_inputs(config, tokenizer, examples):
# add target tokens, remove bos # add target tokens, remove bos
input_ids[i, newline_plus_inputs: newline_plus_inputs + len(target_tokens)] = target_tokens input_ids[i, newline_plus_inputs: newline_plus_inputs + len(target_tokens)] = target_tokens
# add eos token, enforce stopping if we don't truncate # add eos token; ensure generation stops if inputs aren't truncated
# we don't want long code to stop generating if truncated during training # we don't want long code to stop generating if truncated during training
if newline_plus_inputs + len(target_tokens) < max_length: if newline_plus_inputs + len(target_tokens) < max_length:
input_ids[i, newline_plus_inputs + len(target_tokens)] = tokenizer.eos_token_id input_ids[i, newline_plus_inputs + len(target_tokens)] = tokenizer.eos_token_id
@ -57,7 +57,6 @@ def load_data(config, tokenizer):
dataset_path = config["dataset_path"] dataset_path = config["dataset_path"]
if os.path.exists(dataset_path): if os.path.exists(dataset_path):
# check if path is a directory
if os.path.isdir(dataset_path): if os.path.isdir(dataset_path):
files = glob.glob(os.path.join(dataset_path, "*_clean.jsonl")) files = glob.glob(os.path.join(dataset_path, "*_clean.jsonl"))
else: else:
@ -68,7 +67,7 @@ def load_data(config, tokenizer):
dataset = load_dataset("json", data_files=files, split="train") dataset = load_dataset("json", data_files=files, split="train")
else: else:
dataset = load_dataset(dataset_path,split='train') dataset = load_dataset(dataset_path, split="train")
dataset = dataset.train_test_split(test_size=.05, seed=config["seed"]) dataset = dataset.train_test_split(test_size=.05, seed=config["seed"])