Update README.md

README.md

@@ -97,7 +97,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 tokenizer = AutoTokenizer.from_pretrained("Bochkov/emergent-semantics-model-1024-float-335m")
-model = AutoModelForCausalLM.from_pretrained("Bochkov/emergent-semantics-model-1024-float-335m", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("Bochkov/emergent-semantics-model-1024-float-335m", trust_remote_code=True).to('cuda')
 
 inputs = torch.tensor([tokenizer.encode("Question: What is the capital of Japan?\nAnswer:")], dtype=torch.long, device='cuda')
 
@@ -108,6 +108,9 @@ outputs = model.generate(
 )
 print(tokenizer.decode(outputs[0].tolist()))
 
+#Question: What is the capital of Japan?
+#Answer:Tokyo Metropolitan 
+
 ```
 
 ---