diff --git a/CHANGELOG.md b/CHANGELOG.md index 2684f13..c5c590d 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,9 +1,9 @@ # Changelog ## main - * Add --prefix option to load-models * Add get-model-source command +* Use models path as a default prefix (could be overwitten by --prefix flag) ## V0.1.1 diff --git a/rai/cmds.go b/rai/cmds.go index 9e1de1d..8383b65 100644 --- a/rai/cmds.go +++ b/rai/cmds.go @@ -528,11 +528,16 @@ func loadModels(cmd *cobra.Command, args []string) { prefix := action.getString("prefix") models := map[string]io.Reader{} for _, arg := range args[1:] { + if !(cmd.Flags().Lookup("prefix").Changed) { + prefix, _ = filepath.Split(arg) + } + name := filepath.Join(prefix, baseSansExt(arg)) r, err := os.Open(arg) if err != nil { fatal(err.Error()) } + models[name] = r } if engine == "" {