Skip to content

Commit 1e1ec8a

Browse files
Fix Load XXX model message in model loader
1 parent 4029373 commit 1e1ec8a

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

src/main/java/org/beehive/gpullama3/model/loader/Qwen2ModelLoader.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,14 @@ public Qwen2ModelLoader(FileChannel fileChannel, GGUF gguf, int contextLength, b
3535

3636
@Override
3737
public Model loadModel() {
38-
try (var ignored = Timer.log("Load Qwen2 model")) {
39-
Map<String, Object> metadata = gguf.getMetadata();
38+
Map<String, Object> metadata = gguf.getMetadata();
39+
String basename = (String) metadata.get("general.basename");
4040

41+
String modelName = "DeepSeek-R1-Distill-Qwen".equals(basename)
42+
? "DeepSeek-R1-Distill-Qwen"
43+
: "Qwen2.5";
44+
45+
try (var ignored = Timer.log("Load " + modelName + " model")) {
4146
// reuse method of Qwen3
4247
Vocabulary vocabulary = loadQwen3Vocabulary(metadata);
4348
boolean isDeepSeekR1DistillQwen = "DeepSeek-R1-Distill-Qwen".equals(metadata.get("general.basename"));

0 commit comments

Comments
 (0)