Views
No views yet
unsloth/llama-3-8b-bnb-4bit (Llama-3-8B optimizado)databricks/databricks-dolly-15kpeft, transformers y bitsandbytes.1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3import torch
4
5model_id = "unsloth/llama-3-8b-bnb-4bit"
6adapter_id = "TU_USUARIO/TU_MODELO"
7
8tokenizer = AutoTokenizer.from_pretrained(model_id)
9model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float16)
10model = PeftModel.from_pretrained(model, adapter_id)
11
12prompt = "### Inst:\nExplain why the sky is blue.\n### Resp:\n"
13inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
14outputs = model.generate(**inputs, max_new_tokens=100)
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))