|
|
import os
|
|
|
import sys
|
|
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
|
|
|
|
|
from .transformer_utils import get_tokenizer, get_sentence_transformer
|
|
|
try:
|
|
|
from utils.smartHybridAttention import SmartHybridAttention, get_hybrid_attention_config
|
|
|
except ImportError:
|
|
|
try:
|
|
|
from smartHybridAttention import SmartHybridAttention, get_hybrid_attention_config
|
|
|
except ImportError:
|
|
|
print("Warning: Could not import SmartHybridAttention")
|
|
|
SmartHybridAttention = None
|
|
|
get_hybrid_attention_config = None
|
|
|
|
|
|
__all__ = ['get_tokenizer', 'get_sentence_transformer', 'SmartHybridAttention', 'get_hybrid_attention_config']
|
|
|
|