korarishi commited on
Commit
c1ea406
·
verified ·
1 Parent(s): ef89cdd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -47,9 +47,9 @@ print(assistant_response)
47
  from transformers import AutoTokenizer, AutoModelForCausalLM
48
  import torch
49
 
50
- tokenizer = AutoTokenizer.from_pretrained("korarishi1027/rishi-2-2b-it")
51
  model = AutoModelForCausalLM.from_pretrained(
52
- "korarishi1027/rishi-2-2b-it",
53
  device_map="auto",
54
  torch_dtype=torch.bfloat16,
55
  )
 
47
  from transformers import AutoTokenizer, AutoModelForCausalLM
48
  import torch
49
 
50
+ tokenizer = AutoTokenizer.from_pretrained("korarishi/rishi-2-2b-it")
51
  model = AutoModelForCausalLM.from_pretrained(
52
+ "korarishi/rishi-2-2b-it",
53
  device_map="auto",
54
  torch_dtype=torch.bfloat16,
55
  )