lrds-code commited on
Commit
333f34f
·
verified ·
1 Parent(s): 826a6d1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -47,8 +47,8 @@ Keep in mind the limitations of this model. It is a model with 1.1B of trained p
47
  ```python
48
  from transformers import AutoModelForCausalLM, AutoTokenizer
49
 
50
- model = AutoModelForCausalLM.from_pretrained('lrds-code/samba-tiny-llama2-sft')
51
- tokenizer = AutoTokenizer.from_pretrained('lrds-code/samba-tiny-llama2-sft')
52
 
53
  text = 'Pergunta: Como desenvolver habilidades de programação em python?'
54
  inputs = tokenizer(text, return_tensors='pt')
 
47
  ```python
48
  from transformers import AutoModelForCausalLM, AutoTokenizer
49
 
50
+ model = AutoModelForCausalLM.from_pretrained('lrds-code/samba-1.1B')
51
+ tokenizer = AutoTokenizer.from_pretrained('lrds-code/samba-1.1B')
52
 
53
  text = 'Pergunta: Como desenvolver habilidades de programação em python?'
54
  inputs = tokenizer(text, return_tensors='pt')