bugfix
This commit is contained in:
parent
8228cb9ae4
commit
1ff2ee44b5
@ -1,5 +1,17 @@
|
|||||||
import torch
|
import torch
|
||||||
|
|
||||||
|
model_pathMap = {
|
||||||
|
}
|
||||||
|
def llm_register(model_key, Klass):
|
||||||
|
model_pathMap[model_key] = Klass
|
||||||
|
|
||||||
|
def get_llm_class(model_path):
|
||||||
|
for k,klass in model_pathMap.items():
|
||||||
|
if len(model_path.split(k)) > 1:
|
||||||
|
return klass
|
||||||
|
print(f'{model_pathMap=}')
|
||||||
|
return None
|
||||||
|
|
||||||
class BaseReranker:
|
class BaseReranker:
|
||||||
def __init__(self, model_id, **kw):
|
def __init__(self, model_id, **kw):
|
||||||
self.model_id = model_id
|
self.model_id = model_id
|
||||||
|
Loading…
Reference in New Issue
Block a user