Skip to content

Commit 0d1906f

Browse files
fix exaone import (#659)
1 parent 4b10b31 commit 0d1906f

File tree

1 file changed

+8
-4
lines changed

1 file changed

+8
-4
lines changed

awq/models/exaone.py

+8-4
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,14 @@
44
from awq.utils.fused_utils import fuse_qkv
55
from awq.modules.fused.block import LlamaLikeBlock
66
from awq.modules.fused.model import LlamaLikeModel
7-
from transformers.models.exaone.modeling_exaone import (
8-
ExaoneBlock as OldExaoneBlock,
9-
ExaoneForCausalLM as OldExaoneForCausalLM,
10-
)
7+
try:
8+
from transformers.models.exaone.modeling_exaone import (
9+
ExaoneBlock as OldExaoneBlock,
10+
ExaoneForCausalLM as OldExaoneForCausalLM,
11+
)
12+
except:
13+
OldExaoneBlock = None
14+
OldExaoneForCausalLM = None
1115
from awq.modules.fused.norm import FasterTransformerRMSNorm
1216

1317

0 commit comments

Comments
 (0)