Hugging Face's logo

eibeel
/
mixtral_FineTuning_TFG

Model card Files Community
eibeel commited on
Commit
829f82f
·
verified ·
1 Parent(s): e9b6b65

Update MIXTRAL_Fine-Tuning.py

Browse files
Files changed (1) hide show
  1. MIXTRAL_Fine-Tuning.py +1 -1
MIXTRAL_Fine-Tuning.py CHANGED
@@ -44,7 +44,7 @@ from peft import PeftModel, prepare_model_for_kbit_training, LoraConfig
44
  os.environ['HF_TOKEN'] = 'XXXX'
45
 
46
  # Name of the model you want to load
47
- model_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
48
 
49
  try:
50
  # Attempt to load the tokenizer
 
44
  os.environ['HF_TOKEN'] = 'XXXX'
45
 
46
  # Name of the model you want to load
47
+ model_id = "mistralai/Mistral-7B-Instruct-v0.1"
48
 
49
  try:
50
  # Attempt to load the tokenizer