Skip to content

Commit d528460

Browse files
authored
Merge pull request #247 from doringeman/distribution
chore: remove unused NewModelWithConfigArchive
2 parents abad2d8 + d047205 commit d528460

File tree

1 file changed

+0
-34
lines changed
  • pkg/distribution/internal/safetensors

1 file changed

+0
-34
lines changed

pkg/distribution/internal/safetensors/create.go

Lines changed: 0 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -72,40 +72,6 @@ func NewModel(paths []string) (*Model, error) {
7272
}, nil
7373
}
7474

75-
// NewModelWithConfigArchive creates a new safetensors model with a config archive
76-
func NewModelWithConfigArchive(safetensorsPaths []string, configArchivePath string) (*Model, error) {
77-
model, err := NewModel(safetensorsPaths)
78-
if err != nil {
79-
return nil, err
80-
}
81-
82-
// Add config archive layer
83-
if configArchivePath != "" {
84-
// Check if a config archive layer already exists
85-
for _, layer := range model.layers {
86-
mediaType, err := layer.MediaType()
87-
if err == nil && mediaType == types.MediaTypeVLLMConfigArchive {
88-
return nil, fmt.Errorf("model already has a config archive layer")
89-
}
90-
}
91-
92-
configLayer, err := partial.NewLayer(configArchivePath, types.MediaTypeVLLMConfigArchive)
93-
if err != nil {
94-
return nil, fmt.Errorf("create config archive layer from %q: %w", configArchivePath, err)
95-
}
96-
97-
diffID, err := configLayer.DiffID()
98-
if err != nil {
99-
return nil, fmt.Errorf("get config archive layer diffID: %w", err)
100-
}
101-
102-
model.layers = append(model.layers, configLayer)
103-
model.configFile.RootFS.DiffIDs = append(model.configFile.RootFS.DiffIDs, diffID)
104-
}
105-
106-
return model, nil
107-
}
108-
10975
// discoverSafetensorsShards attempts to auto-discover all shards for a given safetensors file
11076
// It looks for the pattern: <name>-XXXXX-of-YYYYY.safetensors
11177
// Returns (nil, nil) for single-file models, (paths, nil) for complete shard sets,

0 commit comments

Comments
 (0)