Update README.md
Browse files
README.md
CHANGED
|
@@ -53,8 +53,8 @@ This model is suitable for:
|
|
| 53 |
```python
|
| 54 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 55 |
|
| 56 |
-
model = AutoModelForCausalLM.from_pretrained("
|
| 57 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
| 58 |
|
| 59 |
prompt = """Question: What is the boiling point of water?\nChoices:\nA. 50°C\nB. 75°C\nC. 90°C\nD. 100°C\nAnswer:"""
|
| 60 |
inputs = tokenizer(prompt, return_tensors="pt")
|
|
|
|
| 53 |
```python
|
| 54 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 55 |
|
| 56 |
+
model = AutoModelForCausalLM.from_pretrained("TechyCode/tinyllama-sciq-lora")
|
| 57 |
+
tokenizer = AutoTokenizer.from_pretrained("TechyCode/tinyllama-sciq-lora")
|
| 58 |
|
| 59 |
prompt = """Question: What is the boiling point of water?\nChoices:\nA. 50°C\nB. 75°C\nC. 90°C\nD. 100°C\nAnswer:"""
|
| 60 |
inputs = tokenizer(prompt, return_tensors="pt")
|