Update README.md
Browse files
README.md
CHANGED
|
@@ -47,6 +47,7 @@ Você pode usar o Bode facilmente com a biblioteca Transformers do HuggingFace.
|
|
| 47 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|
| 48 |
from peft import PeftModel, PeftConfig
|
| 49 |
|
|
|
|
| 50 |
config = PeftConfig.from_pretrained(llm_model)
|
| 51 |
model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, trust_remote_code=True, return_dict=True, load_in_8bit=True, device_map='auto')
|
| 52 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
|
|
|
| 47 |
from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
|
| 48 |
from peft import PeftModel, PeftConfig
|
| 49 |
|
| 50 |
+
llm_model = 'recogna-nlp/bode-7b-alpaca-pt-br'
|
| 51 |
config = PeftConfig.from_pretrained(llm_model)
|
| 52 |
model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, trust_remote_code=True, return_dict=True, load_in_8bit=True, device_map='auto')
|
| 53 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|