Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ollama
Commits
7ffe4573
Commit
7ffe4573
authored
Apr 24, 2024
by
Michael Yang
Browse files
rebase
parent
01811c17
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
15 deletions
+4
-15
convert/mixtral.go
convert/mixtral.go
+3
-14
server/images.go
server/images.go
+1
-1
No files found.
convert/mixtral.go
View file @
7ffe4573
package
convert
package
convert
import
(
import
(
"o
s
"
"
i
o"
"regexp"
"regexp"
"github.com/ollama/ollama/llm"
"github.com/ollama/ollama/llm"
...
@@ -47,7 +47,7 @@ func (m *MixtralModel) LoadVocab() error {
...
@@ -47,7 +47,7 @@ func (m *MixtralModel) LoadVocab() error {
return
nil
return
nil
}
}
func
(
m
*
MixtralModel
)
WriteGGUF
(
)
(
string
,
error
)
{
func
(
m
*
MixtralModel
)
WriteGGUF
(
ws
io
.
WriteSeeker
)
error
{
kv
:=
llm
.
KV
{
kv
:=
llm
.
KV
{
"general.architecture"
:
"llama"
,
"general.architecture"
:
"llama"
,
"general.name"
:
m
.
Name
,
"general.name"
:
m
.
Name
,
...
@@ -81,16 +81,5 @@ func (m *MixtralModel) WriteGGUF() (string, error) {
...
@@ -81,16 +81,5 @@ func (m *MixtralModel) WriteGGUF() (string, error) {
"tokenizer.ggml.add_eos_token"
:
false
,
"tokenizer.ggml.add_eos_token"
:
false
,
}
}
f
,
err
:=
os
.
CreateTemp
(
""
,
"ollama-gguf"
)
return
llm
.
NewGGUFV3
(
m
.
Params
.
ByteOrder
)
.
Encode
(
ws
,
kv
,
m
.
Tensors
)
if
err
!=
nil
{
return
""
,
err
}
defer
f
.
Close
()
mod
:=
llm
.
NewGGUFV3
(
m
.
Params
.
ByteOrder
)
if
err
:=
mod
.
Encode
(
f
,
kv
,
m
.
Tensors
);
err
!=
nil
{
return
""
,
err
}
return
f
.
Name
(),
nil
}
}
server/images.go
View file @
7ffe4573
...
@@ -345,7 +345,7 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, m
...
@@ -345,7 +345,7 @@ func CreateModel(ctx context.Context, name, modelFileDir, quantization string, m
switch
c
.
Name
{
switch
c
.
Name
{
case
"model"
,
"adapter"
:
case
"model"
,
"adapter"
:
var
baseLayers
*
ordered
.
Map
[
*
Layer
,
*
llm
.
GGML
]
var
baseLayers
*
ordered
.
Map
[
*
Layer
,
*
llm
.
GGML
]
if
name
:=
model
.
ParseName
(
c
.
Args
,
""
);
name
.
IsValid
()
{
if
name
:=
model
.
ParseName
(
c
.
Args
);
name
.
IsValid
()
{
baseLayers
,
err
=
parseFromModel
(
ctx
,
name
,
fn
)
baseLayers
,
err
=
parseFromModel
(
ctx
,
name
,
fn
)
if
err
!=
nil
{
if
err
!=
nil
{
return
err
return
err
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment