fix: import path

This commit is contained in:
Zach Nussbaum 2023-05-01 21:39:09 +00:00
parent 0c0a56acab
commit 1b3f18bef2

View File

@ -4,7 +4,7 @@ import torch.nn as nn
from argparse import ArgumentParser from argparse import ArgumentParser
from gpt4all.utils.read import read_config from gpt4all.utils.read import read_config
from accelerate.utils import set_seed from accelerate.utils import set_seed
from gpt4all.utils.data import load_data_for_inference from gpt4all.data.instruction_tuning_dataloader import load_data_for_inference
from gpt4all.utils.distributed_utils import rank0_print from gpt4all.utils.distributed_utils import rank0_print
from tqdm import tqdm from tqdm import tqdm
from datasets import Dataset from datasets import Dataset