baize-chatbot
baize-chatbot copied to clipboard
Features : Can it use 4 bit QLORA ?
from https://huggingface.co/blog/4bit-transformers-bitsandbytes ?
Yes, just modify the utils.py following the example from HF blog:
model = AutoModelForCausalLM.from_pretrained("facebook/opt-350m", load_in_4bit=True, device_map="auto")
We'll update the code after testing.
cool, thanks @JetRunner