Update README.md

README.md

@@ -47,7 +47,7 @@ pip3 install -U bitsandbytes==0.39.0 einops==0.6.1 sentencepiece
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer
 
-model = AutoModelForCausalLM.from_pretrained("azale-ai/DukunLM-7B-V1.0-Uncensored", torch_dtype=torch.float16)
+model = AutoModelForCausalLM.from_pretrained("azale-ai/DukunLM-7B-V1.0-Uncensored", torch_dtype=torch.float16).to("cuda")
 tokenizer = AutoTokenizer.from_pretrained("azale-ai/DukunLM-7B-V1.0-Uncensored")
 streamer = TextStreamer(tokenizer)
 
@@ -94,7 +94,7 @@ _ = model.generate(
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model = AutoModelForCausalLM.from_pretrained("azale-ai/DukunLM-7B-V1.0-Uncensored", torch_dtype=torch.float16)
+model = AutoModelForCausalLM.from_pretrained("azale-ai/DukunLM-7B-V1.0-Uncensored", torch_dtype=torch.float16).to("cuda")
 tokenizer = AutoTokenizer.from_pretrained("azale-ai/DukunLM-7B-V1.0-Uncensored")
 
 instruction_prompt = "Jelaskan mengapa air penting bagi kehidupan manusia."