ciyidogan commited on
Commit
3daccf5
·
verified ·
1 Parent(s): 9a86c53

Update train_lora_mistral.py

Browse files
Files changed (1) hide show
  1. train_lora_mistral.py +1 -1
train_lora_mistral.py CHANGED
@@ -11,7 +11,7 @@ import torch
11
  # === Sabitler ===
12
  START_NUMBER = 0
13
  END_NUMBER = 9
14
- MODEL_NAME = "mistralai/Mistral-7B-Instruct-v0.2"
15
  TOKENIZED_DATASET_ID = "UcsTurkey/turkish-general-culture-tokenized"
16
  ZIP_UPLOAD_REPO = "UcsTurkey/trained-zips"
17
  HF_TOKEN = os.environ.get("HF_TOKEN")
 
11
  # === Sabitler ===
12
  START_NUMBER = 0
13
  END_NUMBER = 9
14
+ MODEL_NAME = "malhajar/Mistral-7B-Instruct-v0.2-turkish"
15
  TOKENIZED_DATASET_ID = "UcsTurkey/turkish-general-culture-tokenized"
16
  ZIP_UPLOAD_REPO = "UcsTurkey/trained-zips"
17
  HF_TOKEN = os.environ.get("HF_TOKEN")