Merge pull request #184 from SecretiveShell/Infinity-Embed-TODO
Complete conditional infinity import TODO
This commit is contained in:
commit
8524999284
1 changed files with 2 additions and 2 deletions
|
|
@ -7,13 +7,13 @@ from typing import List, Optional
|
|||
from common.utils import unwrap
|
||||
|
||||
# Conditionally import infinity to sidestep its logger
|
||||
# TODO: Make this prettier
|
||||
has_infinity_emb: bool = False
|
||||
try:
|
||||
from infinity_emb import EngineArgs, AsyncEmbeddingEngine
|
||||
|
||||
has_infinity_emb = True
|
||||
except ImportError:
|
||||
has_infinity_emb = False
|
||||
pass
|
||||
|
||||
|
||||
class InfinityContainer:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue