mirror of
https://github.com/pese-git/llm-arch-research.git
synced 2026-01-26 22:19:16 +00:00
fix(hf-integration): handle logits as tuple in hf_adapter, convert torch.Tensor to list in hf_tokenizer.decode for decoding compatibility
This commit is contained in:
@@ -99,7 +99,11 @@ class HFGPTAdapter(PreTrainedModel):
|
||||
return_dict = return_dict if return_dict is not None else self.config.use_return_dict
|
||||
|
||||
# Основной forward pass
|
||||
logits = self.llm_model(input_ids)
|
||||
outputs = self.llm_model(input_ids)
|
||||
if isinstance(outputs, tuple):
|
||||
logits = outputs[0]
|
||||
else:
|
||||
logits = outputs
|
||||
|
||||
loss = None
|
||||
if labels is not None:
|
||||
|
||||
Reference in New Issue
Block a user