fix python 3.9 error (#358)

ModelCloud · Aug 13, 2024 · 831e09f · 831e09f
1 parent 25a32e9
commit 831e09f
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