simcop2387/AutoGPTQ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 1 Wiki Activity

make data_utils.py as global utils

Browse source
This commit is contained in:
PanQiWei 2023-04-28 18:08:58 +08:00
parent e189d91004
commit 41564a48db
4 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
auto_gptq/eval_tasks/__init__.py
View file

@ -1,4 +1,3 @@
from . import _utils as eval_task_utils
from .language_modeling_task import * from .language_modeling_task import *
from .sequence_classification_task import * from .sequence_classification_task import *
from .text_summarization_task import * from .text_summarization_task import *

2
auto_gptq/eval_tasks/_base.py
View file

@ -4,8 +4,8 @@ from typing import Any, Dict, List, Optional, Union
import torch import torch
from transformers import PreTrainedTokenizer, PreTrainedModel from transformers import PreTrainedTokenizer, PreTrainedModel
from ._utils.data_utils import get_dataloader
from ..modeling import BaseGPTQForCausalLM from ..modeling import BaseGPTQForCausalLM
from ..utils.data_utils import get_dataloader
class BaseTask: class BaseTask:

0
auto_gptq/utils/__init__.py Normal file
View file

0
auto_gptq/eval_tasks/_utils/data_utils.py → auto_gptq/utils/data_utils.py
View file