Downloads · 30 days
60
8% of all-time downloads
opt-125m-4bit-gptq
seonglae/opt-125m-4bit-gptq
opt-125m-4bit-gptq is a text generation model from seonglae. Use it when you need the model to write or continue text. It is set up for transformers. The card lists the license as mit.
This model should use AutoGPTQ so you need to use auto-gptq
All-time downloads
734
Public
Repo size
486 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors202 MB · 61%
At a glance
- Task
- Text Generation
- Library
- transformers
- License
- mit
- Model type
- opt
- Access
- Public
- Created
- Jul 8, 2023
- Updated
- Jul 19, 2023
- SHA
- 563d486e
Try a prompt
Model card
From the Hugging Face model README
This model should use AutoGPTQ so you need to use auto-gptq
from transformers import AutoTokenizer, pipeline, AutoModelForCausalLM, LlamaForCausalLM, LlamaTokenizer, StoppingCriteria, PreTrainedTokenizerBase
from auto_gptq import AutoGPTQForCausalLM
model_id = 'seonglae/opt-125m-4bit-gptq'
tokenizer = AutoTokenizer.from_pretrained(model_id, use_fast=True)
model = AutoGPTQForCausalLM.from_quantized(
model_id,
model_basename=model_basename,
trust_remote_code=True,
device='cuda:0',
use_triton=False,
use_safetensors=True,
)
pipe = pipeline(
"text-generation",
model=model,
tokenizer=tokenizer,
temperature=0.5,
top_p=0.95,
max_new_tokens=100,
repetition_penalty=1.15,
)
prompt = "USER: Are you AI?\nASSISTANT:"
pipe(prompt)
- Task
- Text Generation
- Library
- transformers
- Type
- opt
- License
- mit
- Created
- Jul 8, 2023
- Updated
- Jul 19, 2023