Update README.md

README.md

@@ -90,7 +90,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("Bochkov/emergent-semantics-model-64-float-272m")
-model = AutoModelForCausalLM.from_pretrained("Bochkov/emergent-semantics-model-64-float-272m", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Bochkov/emergent-semantics-model-64-float-272m", trust_remote_code=True).to('cuda')
 
 inputs = torch.tensor([tokenizer.encode("Question: What is the capital of Japan?\nAnswer:")], dtype=torch.long, device='cuda')
 
@@ -101,6 +101,10 @@ outputs = model.generate(
 )
 print(tokenizer.decode(outputs[0].tolist()))
 
+#Question: What is the capital of Japan?
+#Answer:Japan
+#    </s><|
+
 ```
 
 ---