import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer model_name = "stockmark/Stockmark-2-100B-Instruct" tokenizer = AutoTokenizer.from_pretrained(model_name)model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", torch_dtype="bfloat16") instruction = "自然言語処理とは?"input_ids = tokenizer.apply_chat_template( [{"role": "user", "content": instruction}], add_generation_prompt=True, return_tensors="pt").to(model.device) with torch.inference_mode(): tokens = model.generate( input_ids, max_new_tokens = 512, do_sample = True, temperature = 0.7, top_p = 0.95 ) output = tokenizer.decode(tokens[0], skip_special_tokens=True)print(output)