Update README.md
Browse files
README.md
CHANGED
@@ -22,6 +22,8 @@ extra_gated_description: If you want to learn more about how we process your per
|
|
22 |
import torch
|
23 |
from transformers import pipeline, set_seed
|
24 |
|
|
|
|
|
25 |
pipe = pipeline("text-generation", model=model_path, torch_dtype=torch.bfloat16, device_map="cuda")
|
26 |
messages = [{"role": "user", "content": "what is meaning of life?"}]
|
27 |
outputs = pipe(messages, max_new_tokens=1000, do_sample=True, temperature=0.71, top_k=50, top_p=0.92,repetition_penalty=1)
|
@@ -39,4 +41,4 @@ make the model finely respect guardrails, allowing for deployment in environment
|
|
39 |
|
40 |
## Develop By
|
41 |
|
42 |
-
[email protected]
|
|
|
22 |
import torch
|
23 |
from transformers import pipeline, set_seed
|
24 |
|
25 |
+
model_path = "vicky4s4s/mistral-7b-v2-instruct"
|
26 |
+
|
27 |
pipe = pipeline("text-generation", model=model_path, torch_dtype=torch.bfloat16, device_map="cuda")
|
28 |
messages = [{"role": "user", "content": "what is meaning of life?"}]
|
29 |
outputs = pipe(messages, max_new_tokens=1000, do_sample=True, temperature=0.71, top_k=50, top_p=0.92,repetition_penalty=1)
|
|
|
41 |
|
42 |
## Develop By
|
43 |
|
44 |
+
Vignesh, [email protected]
|