import torchfrom transformers import AutoModelForCausalLM, AutoTokenizerfrom peft import PeftModel base_model_id = "Qwen/Qwen2.5-0.5B-Instruct"adapter_id = "CromIA/think-vetor-0.5b-lora" # Carregar o Tokenizertokenizer = AutoTokenizer.from_pretrained(adapter_id, trust_remote_code=True) # Carregar o Modelo Base com otimização para CPU/GPUdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")dtype = torch.float16 if device.type == "cuda" else torch.float32 model = AutoModelForCausalLM.from_pretrained( base_model_id, torch_dtype=dtype, device_map="auto" if device.type == "cuda" else None, trust_remote_code=True) # Acoplar adaptadores LoRA cognitivosmodel = PeftModel.from_pretrained(model, adapter_id)model.eval() # Definir prompt de testeprompt = "quanto é 432 vezes 78?"messages = [ {"role": "system", "content": "Você é o Think-Vetor 1.5B, um assistente cognitivo híbrido dotado de cadeias de raciocínio de alta fidelidade e raciocínio lógico-matemático."}, {"role": "user", "content": prompt}] formatted_prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)inputs = tokenizer(formatted_prompt, return_tensors="pt").to(device) print("\nRefletindo no Espaço Latente...")with torch.no_grad(): outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.1, do_sample=False) # Decodificargenerated_text = tokenizer.decode(outputs[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True)print("\nResposta do Modelo:", generated_text)