Merge branch 'main' of https://github.com/yuchenlin/LLM-Blender

yuchenlin · Apr 23, 2024 · a38b61c · a38b61c
2 parents e018d7f + de20acb
commit a38b61c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.gitmodules b/.gitmodules
@@ -1,3 +1,6 @@
 [submodule "data/UnifiedFeedback"]
 	path = data/UnifiedFeedback
-	url = https://github.com/jdf-prog/UnifiedFeedback.git
+	url = https://github.com/jdf-prog/UnifiedFeedback
+[submodule "hf_space/LLM-Blender"]
+	path = hf_space/LLM-Blender
+	url = https://huggingface.co/spaces/llm-blender/LLM-Blender
diff --git a/train_ranker.sh b/train_ranker.sh
@@ -15,7 +15,7 @@ dataset="UnifiedFeedback"
 eval_dataset="reward_bench"
 backbone_type="phi" # "deberta" or "roberta"
 backbone_name="microsoft/phi-2" # "microsoft/deberta-v3-large" or "roberta-large"
-n_gpu=4
+n_gpu=8
 ranker="PairRanker" # "PairRanker" or "Summareranker" or "SimCLS"
 candidate_model="" # separted by comma. Empty string for all models
 candidate_decoding_method="" # separted by comma. Empty string for all methods