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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -10,7 +10,7 @@ tags:
10
 
11
  # Rishi-2-2B-IT
12
 
13
- **Model ID:** `korarishi1027/rishi-2-2b-it`
14
 
15
  ## Model Information
16
  Summary description and brief definition of inputs and outputs.
@@ -25,7 +25,7 @@ from transformers import pipeline
25
 
26
  pipe = pipeline(
27
  "text-generation",
28
- model="korarishi1027/rishi-2-2b-it",
29
  model_kwargs={"torch_dtype": torch.bfloat16},
30
  device="cuda", # replace with "mps" to run on a Mac device
31
  )
@@ -74,4 +74,4 @@ outputs = model.generate(**input_ids, max_new_tokens=256)
74
  print(tokenizer.decode(outputs[0]))
75
  ```
76
 
77
- Developed by: [korarishi1027](https://huggingface.co/korarishi1027)
 
10
 
11
  # Rishi-2-2B-IT
12
 
13
+ **Model ID:** `korarishi/rishi-2-2b-it`
14
 
15
  ## Model Information
16
  Summary description and brief definition of inputs and outputs.
 
25
 
26
  pipe = pipeline(
27
  "text-generation",
28
+ model="korarishi/rishi-2-2b-it",
29
  model_kwargs={"torch_dtype": torch.bfloat16},
30
  device="cuda", # replace with "mps" to run on a Mac device
31
  )
 
74
  print(tokenizer.decode(outputs[0]))
75
  ```
76
 
77
+ Developed by: [korarishi](https://huggingface.co/korarishi)