Skip to content

Commit b0f0472

Browse files
QinYuuuuDev Agent
andauthored
fix framework tag recoginization (#605)
Co-authored-by: Dev Agent <[email protected]>
1 parent c798c50 commit b0f0472

File tree

1 file changed

+1
-9
lines changed

1 file changed

+1
-9
lines changed

builder/store/database/repository.go

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -250,15 +250,7 @@ func (r Repository) Archs() (archs []string) {
250250
func (r Repository) Format() string {
251251
for _, tag := range r.Tags {
252252
if tag.Category == "framework" {
253-
if tag.Name == "gguf" {
254-
return tag.Name
255-
}
256-
if tag.Name == "onnx" {
257-
return tag.Name
258-
}
259-
if tag.Name == "safetensors" {
260-
return tag.Name
261-
}
253+
return tag.Name
262254
}
263255
}
264256
//handle some old repo has no gguf tag

0 commit comments

Comments
 (0)