import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfigfrom peft import PeftModel base_model_name = "Qwen/Qwen2.5-Coder-7B-Instruct"adapter_id = "Ninitje/InvariantMind-Worker-7B" bnb_config = BitsAndBytesConfig( load_in_4bit=True, bnb_4bit_compute_dtype=torch.bfloat16, bnb_4bit_quant_type="nf4") tokenizer = AutoTokenizer.from_pretrained(base_model_name)base_model = AutoModelForCausalLM.from_pretrained( base_model_name, quantization_config=bnb_config, device_map="auto")model = PeftModel.from_pretrained(base_model, adapter_id) prompt = "Write an optimized, vectorized Python implementation using NumPy to simulate N coupled Kuramoto oscillators and compute order parameter R(t)."messages = [{"role": "user", "content": prompt}]inputs = tokenizer(tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True), return_tensors="pt").to("cuda") with torch.no_grad(): outputs = model.generate(**inputs, max_new_tokens=600, temperature=0.6)print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))