[BugFix]:Text Classification pipeline,ValueError-trying to set a tensor of incorrect shape in weight (#3592)

This commit is contained in:
Yeshwanth N 2024-11-12 15:57:04 +05:30 коммит произвёл GitHub
Родитель 92b77d33e5
Коммит 863be53185
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -9,7 +9,7 @@ sentencepiece==0.1.99
transformers==4.44.0
datasets==2.17.1
optimum==1.17.1
accelerate==0.33.0
accelerate==1.1.0
diffusers==0.26.3
onnxruntime==1.17.3
rouge-score==0.1.2