Update README.md (#48)

Author: eryk-mazus

README.md

@@ -47,7 +47,7 @@ checkpoint = "bigcode/starcoder"
 device = "cuda" # for GPU usage or "cpu" for CPU usage
 
 tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-# to save memory consider using fp16 or bf16 by specifying torch.dtype=torch.float16 for example
+# to save memory consider using fp16 or bf16 by specifying torch_dtype=torch.float16 for example
 model = AutoModelForCausalLM.from_pretrained(checkpoint).to(device)
 
 inputs = tokenizer.encode("def print_hello_world():", return_tensors="pt").to(device)