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
90c49bed
"examples/controlnet/train_controlnet.py" did not exist on "a7ca03aa85f94574f06576d2155b3ec061fe8d63"
Commit
90c49bed
authored
Oct 18, 2023
by
Arne Müller
Browse files
moved removal of leading space into Predict
parent
5dc0cff4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
1 deletion
+6
-1
llm/llama.go
llm/llama.go
+6
-1
No files found.
llm/llama.go
View file @
90c49bed
...
...
@@ -464,6 +464,11 @@ func (llm *llama) Predict(ctx context.Context, prevContext []int, prompt string,
return
err
}
// Remove first leading space from prevConvo if present
if
len
(
prevConvo
)
>
0
&&
prevConvo
[
0
]
==
' '
{
prevConvo
=
prevConvo
[
1
:
]
}
var
nextContext
strings
.
Builder
nextContext
.
WriteString
(
prevConvo
)
nextContext
.
WriteString
(
prompt
)
...
...
@@ -666,7 +671,7 @@ func (llm *llama) Decode(ctx context.Context, tokens []int) (string, error) {
}
// decoded content contains a leading whitespace
decoded
.
Content
,
_
=
strings
.
CutPrefix
(
decoded
.
Content
,
"
"
)
decoded
.
Content
,
_
=
strings
.
CutPrefix
(
decoded
.
Content
,
""
)
return
decoded
.
Content
,
nil
}
...
...
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