vicky4s4s commited on
Commit
0a052eb
·
verified ·
1 Parent(s): ec5582d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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], https://www.linkedin.com/in/vicky-s-41135319a/overlay/create-post/
 
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]