Skip to content

Move labels to the same device as logits for LlamaForSequenceClassification and Blip2#22596

Merged
sgugger merged 5 commits into
huggingface:mainfrom
xssChauhan:llama_seqclassification_parllelism
Apr 7, 2023
Merged

Move labels to the same device as logits for LlamaForSequenceClassification and Blip2#22596
sgugger merged 5 commits into
huggingface:mainfrom
xssChauhan:llama_seqclassification_parllelism

Commits

Commits on Apr 6, 2023