Downloads · 30 days
0
ArnabAr/SFT_Lora_Mistral_700_Data_V1
SFT_Lora_Mistral_700_Data_V1 is a machine learning model from ArnabAr. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
- Developed by: ArnabAr - License: apache-2.0 - Finetuned from model : unsloth/mistral-7b-bnb-4bit
Downloads · 30 days
0
Access
Public
Updated Dec 16, 2024
Repo size
168 MB
Likes
0
Public
Click a slice to open those files.
.safetensors168 MB · 98%
From the Hugging Face model README
from unsloth import FastLanguageModel
max_seq_length = 4096 # Choose any! We auto support RoPE Scaling internally!
dtype = None # None for auto detection. Float16 for Tesla T4, V100, Bfloat16 for Ampere+
load_in_4bit = True # Use 4bit quantization to reduce memory usage. Can be False.
model, tokenizer = FastLanguageModel.from_pretrained( model_name = "ArnabAr/SFT_Lora_Mistral_700_Data_V1", # YOUR MODEL YOU USED FOR TRAINING max_seq_length = max_seq_length, dtype = dtype, load_in_4bit = load_in_4bit, )
FastLanguageModel.for_inference(model) # Enable native 2x faster inference
messages = [ {"from": "human", "value": " Ask you queries. " }, ]
input_text = messages[0]["value"]
inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
output = model.generate( input_ids=inputs["input_ids"], max_new_tokens=max_seq_length, use_cache=True ) response = tokenizer.decode(output[0], skip_special_tokens=True)
print(response)
This mistral model was trained 2x faster with Unsloth and Huggingface's TRL library.