fix python 3.9 error

ModelCloud · Aug 13, 2024 · bd78096 · bd78096
1 parent 0390925
commit bd78096
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/gptqmodel/models/auto.py b/gptqmodel/models/auto.py
@@ -1,3 +1,4 @@
+from __future__ import annotations
 from typing import Dict, List, Optional, Union
 
 from ..utils import BACKEND

diff --git a/gptqmodel/models/base.py b/gptqmodel/models/base.py
@@ -1,3 +1,4 @@
+from __future__ import annotations
 import copy
 import json
 import logging

diff --git a/gptqmodel/utils/model.py b/gptqmodel/utils/model.py
@@ -1,3 +1,4 @@
+from __future__ import annotations
 import functools
 import hashlib
 import json