update tests

This commit is contained in:
Michael Yang
2025-09-22 12:57:30 -07:00
parent b846eacf42
commit ffaf2e7916
11 changed files with 323 additions and 276 deletions

View File

@@ -88,10 +88,10 @@ func TestConvertFromSafetensors(t *testing.T) {
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
// Create the minimum required file map for convertFromSafetensors
files := map[string]string{
tt.filePath: model,
"config.json": config,
"tokenizer.json": tokenizer,
files := []api.File{
{Name: tt.filePath, Digest: model},
{Name: "config.json", Digest: config},
{Name: "tokenizer.json", Digest: tokenizer},
}
_, err := convertFromSafetensors(files, nil, false, func(resp api.ProgressResponse) {})