This commit is contained in:
narawat lamaiin
2024-07-16 18:18:08 +07:00
parent 1ef696f206
commit f5a9454aa3
2 changed files with 4 additions and 2 deletions

View File

@@ -42,7 +42,7 @@ function text2textInstructLLM(prompt::String)
:payload=> Dict(
:text=> prompt,
:kwargs=> Dict(
:max_tokens=> 512,
:max_tokens=> 1024,
:stop=> ["<|eot_id|>"],
:temperature=> 0.2,
)
@@ -75,7 +75,9 @@ function main()
userinput = "Hello, I would like a get a bottle of wine."
for i in 1:10
response = YiemAgent.conversation(a, Dict(:text=> userinput))
println("")
println("--> assistant response: \n", response)
println("")
println("--> user input:")
userinput = readline()
end