Skip to content

Commit

Permalink
Add recipes/A5000_24GB_x8/Mistral-7B-v0.1/i18n-ja-wikipedia-sakura.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
yuiseki committed May 5, 2024
1 parent 69112cb commit 2010891
Showing 1 changed file with 18 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
target_task: tasks/i18n/ja.md
base_model_id: yuiseki/Mistral-7B-v0.1-ja-wikipedia-v0.1
model_name: Mistral-7B-v0.1-ja-wikipedia-sakura-v0.1
output_base_dir: output
dataset_id: saldra/sakura_japanese_dataset
dataset_context_field_name: input
dataset_input_field_name: instruction
dataset_output_field_name: output
dataset_filter_field_value: 0
dataset_train_split_seed: 42
dataset_train_split_test_size: 0.2
lora_r: 8
lora_alpha: 16
lora_dropout: 0.05
train_claim_gpu_num: 2
train_per_device_train_batch_size: 2
train_gradient_accumulation_steps: 16
train_num_train_epochs: 4

0 comments on commit 2010891

Please sign in to comment.