Update README.md
Browse files
README.md
CHANGED
@@ -92,9 +92,10 @@ tokenizer.padding_side = "right"
|
|
92 |
```python
|
93 |
# Run text generation pipeline with the merged model
|
94 |
prompt = """
|
95 |
-
|
|
|
96 |
pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer,do_sample = True, temperature = 0.9, max_length=200)
|
97 |
response = pipe(f"<s>[INST] {prompt} [/INST]")
|
98 |
-
print(response)
|
99 |
|
100 |
```
|
|
|
92 |
```python
|
93 |
# Run text generation pipeline with the merged model
|
94 |
prompt = """
|
95 |
+
i have a neck pain since 2 days .
|
96 |
+
"""
|
97 |
pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer,do_sample = True, temperature = 0.9, max_length=200)
|
98 |
response = pipe(f"<s>[INST] {prompt} [/INST]")
|
99 |
+
print(response[0]['generated_text'])
|
100 |
|
101 |
```
|