Update README.md

README.md

@@ -35,7 +35,7 @@ model = AutoModelForCausalLM.from_pretrained(
 )
 
 if torch.cuda.is_available():
-    model = model.to("cuda")
+  model = model.to("cuda")
 
 prompt="""### Instruction:
 光の三原色は？
@@ -44,12 +44,12 @@ prompt="""### Instruction:
 
 input_ids = tokenizer(prompt, return_tensors="pt").to(model.device)
 outputs = model.generate(
-    **input_ids,
-    max_new_tokens=512,
-    do_sample=True,
-    top_p=0.95,
-    temperature=0.1,
-    repetition_penalty=1.0,
+  **input_ids,
+  max_new_tokens=512,
+  do_sample=True,
+  top_p=0.95,
+  temperature=0.1,
+  repetition_penalty=1.0,
 )
 print(tokenizer.decode(outputs[0]))
 ```