import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer model_name = "iamtatsuki05/Llama-JP-0.5B-init"model_kwargs = { "torch_dtype": torch.bfloat16, "attn_implementation": "flash_attention_2", "device_map": "auto",}tokenizer = AutoTokenizer.from_pretrained(model_name)model = AutoModelForCausalLM.from_pretrained(model_name, **model_kwargs) prompt = "ちいかわのハチワレは"inputs = tokenizer(prompt, return_tensors="pt").to(model.device)output = model.generate( **inputs, max_new_tokens=256, temperature=0.8, top_p=0.9, do_sample=True,)print(tokenizer.decode(output[0], skip_special_tokens=True))