from transformers import AutoTokenizer, AutoModelForCausalLM tokenizer = AutoTokenizer.from_pretrained("khazarai/BioGenesis-ToT")model = AutoModelForCausalLM.from_pretrained( "khazarai/BioGenesis-ToT", device_map={"": 0}) question = """Describe the composition of the plasma membrane and explain how its structure relates to its function of selective permeability.""" messages = [ {"role" : "user", "content" : question}]text = tokenizer.apply_chat_template( messages, tokenize = False, add_generation_prompt = True, enable_thinking = True,) from transformers import TextStreamer_ = model.generate( **tokenizer(text, return_tensors = "pt").to("cuda"), max_new_tokens = 2200, temperature = 0.6, top_p = 0.95, top_k = 20, streamer = TextStreamer(tokenizer, skip_prompt = True),)