import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer model_name = "elyza/ELYZA-Shortcut-1.0-Qwen-7B"prompt = "仕事の熱意を取り戻すためのアイデアを5つ挙げてください。" tokenizer = AutoTokenizer.from_pretrained(model_name)model = AutoModelForCausalLM.from_pretrained( model_name, torch_dtype="auto", device_map="auto",)model.eval()messages = [{"role": "user", "content": prompt}]input_text = tokenizer.apply_chat_template( messages, tokenize=False, add_generation_prompt=True)token_ids = tokenizer.encode( input_text, add_special_tokens=False, return_tensors="pt")with torch.no_grad(): output_ids = model.generate( token_ids.to(model.device), max_new_tokens=8192, do_sample=True, temperature=0.6, top_p=0.95, )output = tokenizer.decode( output_ids.tolist()[0][token_ids.size(1):], skip_special_tokens=True)print(output)