Update README.md
Browse files
README.md
CHANGED
|
@@ -28,7 +28,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 28 |
|
| 29 |
tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
|
| 30 |
model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
|
| 31 |
-
model.half()
|
| 32 |
|
| 33 |
inputs = tokenizer("What's your mood today?", return_tensors="pt")
|
| 34 |
tokens = model.generate(
|
|
|
|
| 28 |
|
| 29 |
tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
|
| 30 |
model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-tuned-alpha-7b")
|
|
|
|
| 31 |
|
| 32 |
inputs = tokenizer("What's your mood today?", return_tensors="pt")
|
| 33 |
tokens = model.generate(
|