Update README.md

README.md

@@ -14,19 +14,13 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 torch_dtype = torch.bfloat16 if torch.cuda.is_available() and hasattr(torch.cuda, "is_bf16_supported") and torch.cuda.is_bf16_supported() else torch.float16
-
 model = AutoModelForCausalLM.from_pretrained("Calvin-Xu/FLFL", device_map="auto", torch_dtype=torch_dtype)
 tokenizer = AutoTokenizer.from_pretrained("Calvin-Xu/FLFL")
 
 prompt_template = """[INST] {instruction}\n{input}\n[/INST]\n"""
 sentence = "国境の長いトンネルを抜けると雪国であった"
 
-inputs = tokenizer(
-    prompt_template.format(
-        instruction="次の文に正確に振り仮名を付けてください", input=sentence
-    ),
-    return_tensors="pt",
-).to(model.device)
+inputs = tokenizer(prompt_template.format(instruction="次の文に正確に振り仮名を付けてください", input=sentence), return_tensors="pt").to(model.device)
 with torch.no_grad():
     tokens = model.generate(**inputs, max_new_tokens=512, do_sample=False)