sunitha-ravi commited on
Commit
0747f1a
·
verified ·
1 Parent(s): f36386a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md CHANGED
@@ -63,6 +63,29 @@ Your output should be in JSON FORMAT with the keys "REASONING" and "SCORE":
63
 
64
  The model will output the score as 'PASS' if the answer is faithful to the document or FAIL if the answer is not faithful to the document.
65
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
66
 
67
  ## Training Details
68
 
 
63
 
64
  The model will output the score as 'PASS' if the answer is faithful to the document or FAIL if the answer is not faithful to the document.
65
 
66
+ ## Inference
67
+
68
+ To run inference, you can use HF pipeline:
69
+
70
+ ```
71
+ pipe = pipeline(
72
+ "text-generation",
73
+ model="PatronusAI/Llama-3-Patronus-Lynx-8B-Instruct",
74
+ max_new_tokens=600,
75
+ device="cuda",
76
+ return_full_text=False
77
+ )
78
+
79
+ messages = [
80
+ {"role": "user", "content": prompt},
81
+ ]
82
+
83
+ result = pipe(messages)
84
+ print(result[0]['generated_text'])
85
+
86
+ ```
87
+
88
+ Since the model is trained in chat format, ensure that you pass the prompt as a user message.
89
 
90
  ## Training Details
91