YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

base_model = "mistralai/Mistral-7B-v0.1" dataset_name = "KasparZ/HITL-2"

load_in_4bit= True, bnb_4bit_quant_type= "nf4", bnb_4bit_compute_dtype= torch.bfloat16, bnb_4bit_use_double_quant= False,

device_map="auto", trust_remote_code=True,

lora_alpha=16, lora_dropout=0.1, r=64, bias="none", task_type="CAUSAL_LM", target_modules=["q_proj", "k_proj", "v_proj", "o_proj","gate_proj"]

model.gradient_checkpointing_enable()

tokenizer.padding_side = 'right' tokenizer.add_eos_token = True

num_train_epochs=4,
per_device_train_batch_size=4,
gradient_accumulation_steps=1,
optim="paged_adamw_32bit",
save_steps=50,
logging_steps=1,
learning_rate=2e-4,
weight_decay=0.001,
fp16=False,
bf16=False,
max_grad_norm=0.3,
max_steps=-1,
warmup_ratio=0.03,
group_by_length=True,
lr_scheduler_type="constant",
Downloads last month
14
Safetensors
Model size
7B params
Tensor type
F16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support