mirror of https://github.com/hpcaitech/ColossalAI
13 lines
323 B
Python
13 lines
323 B
Python
|
try:
|
||
|
|
import torch_int
|
||
|
|
|
||
|
|
HAS_TORCH_INT = True
|
||
|
|
except ImportError:
|
||
|
|
HAS_TORCH_INT = False
|
||
|
|
raise ImportError(
|
||
|
|
"Not install torch_int. Please install torch_int from https://github.com/Guangxuan-Xiao/torch-int"
|
||
|
|
)
|
||
|
|
|
||
|
|
if HAS_TORCH_INT:
|
||
|
|
from .llama import LLamaSmoothquantAttention, LlamaSmoothquantMLP
|