Update README.md
Browse files
README.md
CHANGED
@@ -10,7 +10,7 @@ tags:
|
|
10 |
|
11 |
# Rishi-2-2B-IT
|
12 |
|
13 |
-
**Model ID:** `
|
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="
|
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: [
|
|
|
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)
|