license: apache-2.0
Eric Hartford's WizardLM-Uncensored-Falcon-7B GPTQ
This repo contains an experimantal GPTQ 4bit model for Eric Hartford's WizardLM-Uncensored-Falcon-7B.
It is the result of quantising to 4bit using AutoGPTQ.
Repositories available
- 4-bit GPTQ model for GPU inference
- Eric's unquantised bf16 model in pytorch format, for GPU inference and for further conversions
Prompt template
Prompt format is WizardLM:
What is a falcon? Can I keep one as a pet?
### Response:
EXPERIMENTAL
Please note this is an experimental GPTQ model. Support for it is currently quite limited.
It is also expected to be SLOW. This is currently unavoidable, but is being looked at.
AutoGPTQ
AutoGPTQ 0.2.0 is required: pip install auto-gptq
AutoGPTQ provides pre-compiled wheels for Windows and Linux, with CUDA toolkit 11.7 or 11.8.
If you are running CUDA toolkit 12.x, you will need to compile your own by following these instructions:
git clone https://github.com/PanQiWei/AutoGPTQ
cd AutoGPTQ
pip install .
These manual steps will require that you have the Nvidia CUDA toolkit installed.
text-generation-webui
There is provisional AutoGPTQ support in text-generation-webui.
This requires text-generation-webui as of commit 204731952ae59d79ea3805a425c73dd171d943c3.
So please first update text-genration-webui to the latest version.
How to download and use this model in text-generation-webui
- Launch text-generation-webui with the following command-line arguments:
--autogptq --trust-remote-code
- Click the Model tab.
- Under Download custom model or LoRA, enter
TheBloke/WizardLM-Uncensored-Falcon-7B-GPTQ
. - Click Download.
- Wait until it says it's finished downloading.
- Click the Refresh icon next to Model in the top left.
- In the Model drop-down: choose the model you just downloaded,
WizardLM-Uncensored-Falcon-7B-GPTQ
. - Once it says it's loaded, click the Text Generation tab and enter a prompt!
About trust_remote_code
Please be aware that this command line argument causes Python code provided by Falcon to be executed on your machine.
This code is required at the moment because Falcon is too new to be supported by Hugging Face transformers. At some point in the future transformers will support the model natively, and then trust_remote_code
will no longer be needed.
In this repo you can see two .py
files - these are the files that get executed. They are copied from the base repo at Falcon-7B-Instruct.
Simple Python example code
To run this code you need to install AutoGPTQ and einops:
pip install auto-gptq
pip install einops
You can then run this example code:
import torch
from transformers import AutoTokenizer
from auto_gptq import AutoGPTQForCausalLM
# Download the model from HF and store it locally, then reference its location here:
quantized_model_dir = "/path/to/TheBloke_WizardLM-Uncensored-Falcon-7B-GPTQ"
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, use_fast=False)
model = AutoGPTQForCausalLM.from_quantized(quantized_model_dir, device="cuda:0", use_triton=False, use_safetensors=True, torch_dtype=torch.float32, trust_remote_code=True)
prompt = "Write a story about llamas"
prompt_template = f"### Instruction: {prompt}\n### Response:"
tokens = tokenizer(prompt_template, return_tensors="pt").to("cuda:0").input_ids
output = model.generate(input_ids=tokens, max_new_tokens=100, do_sample=True, temperature=0.8)
print(tokenizer.decode(output[0]))
Provided files
gptq_model-4bit-64g.safetensors
This will work with AutoGPTQ as of commit 3cb1bf5
(3cb1bf5a6d43a06dc34c6442287965d1838303d3
)
It was created with groupsize 64 to give higher inference quality, and without desc_act
(act-order) to increase inference speed.
gptq_model-4bit-64g.safetensors
- Works only with latest AutoGPTQ CUDA, compiled from source as of commit
3cb1bf5
- At this time it does not work with AutoGPTQ Triton, but support will hopefully be added in time.
- Works with text-generation-webui using
--autogptq --trust_remote_code
- At this time it does NOT work with one-click-installers
- Does not work with any version of GPTQ-for-LLaMa
- Parameters: Groupsize = 64. No act-order.
- Works only with latest AutoGPTQ CUDA, compiled from source as of commit
Discord
For further support, and discussions on these models and AI in general, join us at: TheBloke AI's Discord server
Thanks, and how to contribute.
Thanks to the chirper.ai team!
I've had a lot of people ask if they can contribute. I enjoy providing models and helping people, and would love to be able to spend even more time doing it, as well as expanding into new projects like fine tuning/training.
If you're able and willing to contribute, it'd be most gratefully received and will help me to keep providing models, and work on new AI projects.
Donaters will get priority support on any and all AI/LLM/model questions, plus other benefits.
- Patreon: https://patreon.com/TheBlokeAI
- Ko-Fi: https://ko-fi.com/TheBlokeAI
Patreon special mentions: Aemon Algiz; Johann-Peter Hartmann; Talal Aujan; Jonathan Leane; Illia Dulskyi; Khalefa Al-Ahmad; senxiiz; Sebastain Graf; Eugene Pentland; Nikolai Manek; Luke Pendergrass.
Thank you to all my generous patrons and donaters.
✨ Original model card: Eric Hartford's WizardLM-Uncensored-Falcon-7B
This is WizardLM trained on top of tiiuae/falcon-7b, with a subset of the dataset - responses that contained alignment / moralizing were removed. The intent is to train a WizardLM that doesn't have alignment built-in, so that alignment (of any sort) can be added separately with for example with a RLHF LoRA.
Shout out to the open source AI/ML community, and everyone who helped me out.
Note: An uncensored model has no guardrails. You are responsible for anything you do with the model, just as you are responsible for anything you do with any dangerous object such as a knife, gun, lighter, or car. Publishing anything this model generates is the same as publishing it yourself. You are responsible for the content you publish, and you cannot blame the model any more than you can blame the knife, gun, lighter, or car for what you do with it.
Prompt format is Wizardlm.
What is a falcon? Can I keep one as a pet?
### Response: