sunitha-ravi
commited on
Update README.md
Browse files
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 |
|