jon-tow commited on
Commit
99be347
·
1 Parent(s): 83e3e96

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -1
README.md CHANGED
@@ -21,7 +21,6 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
21
 
22
  tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-base-alpha-7b")
23
  model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-base-alpha-7b")
24
- model.half()
25
 
26
  inputs = tokenizer("What's your mood today?", return_tensors="pt")
27
  tokens = model.generate(
 
21
 
22
  tokenizer = AutoTokenizer.from_pretrained("StabilityAI/stablelm-base-alpha-7b")
23
  model = AutoModelForCausalLM.from_pretrained("StabilityAI/stablelm-base-alpha-7b")
 
24
 
25
  inputs = tokenizer("What's your mood today?", return_tensors="pt")
26
  tokens = model.generate(