Pentru a rula inferența, mesajul utilizatorului trebuie transformat în tensori PyTorch plasați pe dispozitivul GPU (cuda):
FastLanguageModel.for_inference(model)
prompt = "Explică algoritmul LoRA pe scurt."
inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
outputs = model.generate(
**inputs,
max_new_tokens=128,
use_cache=True, # Activează KV cache pentru viteză
)
rezultat = tokenizer.batch_decode(outputs, skip_special_tokens=True)
print(rezultat[0])
Parametrul use_cache=True este critic pentru a preveni recalcularea atenției pe tokenii anterior generați.
Întrebări & Exemple practice:
Apelarea Metodei Native model.generate cu Tensori PyTorchAi o întrebare despre Apelarea Metodei Native model.generate cu Tensori PyTorch sau ai descoperit un truc, prompt sau fragment de cod util? Distribuie-l cu ceilalți membri ai comunității!
Fii primul care pune o întrebare sau distribuie un fragment de cod / prompt!
Nu au fost adăugate încă întrebări sau resurse pentru această pagină. Dacă ai testat un prompt sau ai un snippet util, adaugă-l mai jos pentru comunitate.