22-hours / cabrita

Finetuning InstructLLaMA with portuguese data
Apache License 2.0
558 stars 68 forks source link

Alteração nos imports LLaMAForCausalLM e LLaMATokenizer #14

Open natansr opened 1 year ago

natansr commented 1 year ago

acho que no código do notebook disponibilizado está com os nomes antigos das classes

from transformers import LLaMATokenizer, LLaMAForCausalLM

A nova atualização colocou o LLaMa como minusculo. Ficando assim:

from transformers import AutoTokenizer, AutoConfig, LlamaForCausalLM, LlamaTokenizer