From 8eddac305857a9ecc19fb2047d724ec0f1c04dd9 Mon Sep 17 00:00:00 2001 From: Ishaan Sehgal Date: Tue, 26 Mar 2024 13:32:04 -0700 Subject: [PATCH] fix: Update Model Tags (#311) Signed-off-by: Ishaan Sehgal --- presets/models/falcon/model.go | 8 ++++---- presets/models/mistral/model.go | 4 ++-- presets/models/phi/model.go | 2 +- 3 files changed, 7 insertions(+), 7 deletions(-) diff --git a/presets/models/falcon/model.go b/presets/models/falcon/model.go index 7501dce23..863a2fb52 100644 --- a/presets/models/falcon/model.go +++ b/presets/models/falcon/model.go @@ -37,10 +37,10 @@ var ( PresetFalcon40BInstructModel = PresetFalcon40BModel + "-instruct" PresetFalconTagMap = map[string]string{ - "Falcon7B": "0.0.3", - "Falcon7BInstruct": "0.0.3", - "Falcon40B": "0.0.3", - "Falcon40BInstruct": "0.0.3", + "Falcon7B": "0.0.4", + "Falcon7BInstruct": "0.0.4", + "Falcon40B": "0.0.5", + "Falcon40BInstruct": "0.0.5", } baseCommandPresetFalcon = "accelerate launch" diff --git a/presets/models/mistral/model.go b/presets/models/mistral/model.go index 7089eafb6..910e2da83 100644 --- a/presets/models/mistral/model.go +++ b/presets/models/mistral/model.go @@ -27,8 +27,8 @@ var ( PresetMistral7BInstructModel = PresetMistral7BModel + "-instruct" PresetMistralTagMap = map[string]string{ - "Mistral7B": "0.0.3", - "Mistral7BInstruct": "0.0.3", + "Mistral7B": "0.0.4", + "Mistral7BInstruct": "0.0.4", } baseCommandPresetMistral = "accelerate launch" diff --git a/presets/models/phi/model.go b/presets/models/phi/model.go index 2e54dce38..32df386cc 100644 --- a/presets/models/phi/model.go +++ b/presets/models/phi/model.go @@ -22,7 +22,7 @@ var ( PresetPhi2Model = "phi-2" PresetPhiTagMap = map[string]string{ - "Phi2": "0.0.2", + "Phi2": "0.0.3", } baseCommandPresetPhi = "accelerate launch"