mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2026-04-20 22:13:43 +00:00
exllamav3: Implement the logits function for /v1/internal/logits
This commit is contained in:
parent
218dc01b51
commit
7f06aec3a1
3 changed files with 46 additions and 5 deletions
|
|
@ -104,8 +104,7 @@ def ExLlamav3_HF_loader(model_name):
|
|||
def ExLlamav3_loader(model_name):
|
||||
from modules.exllamav3 import Exllamav3Model
|
||||
|
||||
model = Exllamav3Model.from_pretrained(model_name)
|
||||
tokenizer = model.tokenizer
|
||||
model, tokenizer = Exllamav3Model.from_pretrained(model_name)
|
||||
return model, tokenizer
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue