From 1b3f18bef262340a03998057849747362bc680cd Mon Sep 17 00:00:00 2001 From: Zach Nussbaum Date: Mon, 1 May 2023 21:39:09 +0000 Subject: [PATCH] fix: import path --- gpt4all/inference/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gpt4all/inference/inference.py b/gpt4all/inference/inference.py index 2096819c..c69e50bf 100644 --- a/gpt4all/inference/inference.py +++ b/gpt4all/inference/inference.py @@ -4,7 +4,7 @@ import torch.nn as nn from argparse import ArgumentParser from gpt4all.utils.read import read_config 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 tqdm import tqdm from datasets import Dataset