地平線まで行ってくる。

記録あるいは忘備録。時には検討事項。

AI対話:会話データSODAで鍛えたCOSMOモデルを試す。

自然な会話ができるChatGPTが話題ですが、SODA*1という、データセットで学習させたCOSMOというモデルがなかな優秀だというのをtwitterで見かけたので試してみる。

 

github.com

 

会話の状況は「最後の一葉」を参考にして投入した。

instructionは不要らしいが、人物関係が分かりやすいかと書いてみる。

situation = "When Sue awoke from an hour's sleep the next morning she found Johnsy with dull, wide-open eyes staring at the drawn green shade."
instruction = "Johnsy is very ill and weak and the fever has left her mind morbid and full of strange fancies. "

最初の会話の言葉として

  "Try to sleep."

に続く会話を生成させてみる。その結果です。

I can't. My mind won't stop racing.
You need to rest. You're not well enough to think clearly right now.
But what if I don't get better? What if this fever kills me?
Johnsy, you're going to be fine. I'm here with you and we'll get through this together.
I'm scared, Sue.
It's okay to be scared. Just try to focus on getting better and I'll be here with you every step of the way.
Alright.
I love you.
I love you too.
What did I tell you? About being scared?
Yes.
It's okay to be scared. We all are when we're faced with a scary situation. But you have to remember that I'm here with you and we'll get through this together, alright?
Okay, Sue.
Good girl. Now get some rest, okay? I'll check on you later.
All right. Love you.
Love you too.
Hey, Sue.
Hey, Johnsy. How are you feeling?
I'm feeling a little better today. The fever seems to be breaking a little bit and my mind is starting to focus more clearly.
That's great! You're finally starting to feel better. Just make sure you get some more rest and we'll talk soon, okay?
Okay. Thanks, Sue.
Of course, sweetie. I'll be right here with you the whole time."
I'm sorry I was so out of it yesterday. I don't know what came over me.
It's okay, Johnsy. I know you've been through a lot lately with your mom being sick and all. You just need to take it easy for a while and let your body heal. We'll get through this together, alright?
Alright, Sue. Thanks for being here with me. I love you.
I love you too, Johnsy."
Hey, Sue. Thanks for being here with me today.
Of course, sweetie. I told you I would be here with you and I meant it."
What if I don't get better? What if this fever kills me?

世間話しつつ英会話のお勉強の相手くらいは十分できそうなレベル。

何度か生成した会話にはモンスターが襲ってきた夢をみたりした状況も出てきて楽しめた(笑)。