Skip to content

Comments

Handle output of huggingface models with optional logits attribute#45

Open
Touka20 wants to merge 1 commit intonoahgolmant:masterfrom
Touka20:supporting-huggingface-models
Open

Handle output of huggingface models with optional logits attribute#45
Touka20 wants to merge 1 commit intonoahgolmant:masterfrom
Touka20:supporting-huggingface-models

Conversation

@Touka20
Copy link

@Touka20 Touka20 commented Oct 7, 2025

For huggingface models usually we need to get "logits". The output itself cannot be used for loss computation.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant