Skip to content

Commit 3ed43f8

Browse files
committed
Update StableDiffusionResponder.
1 parent 32e4b60 commit 3ed43f8

1 file changed

Lines changed: 2 additions & 4 deletions

File tree

src/Automation/Responders/StableDiffusionResponder.cs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
using Discord;
22
using Estranged.Automation.Events;
33
using Microsoft.Extensions.AI;
4-
using Estranged.Automation.Events;
5-
using Microsoft.Extensions.AI;
64
using Microsoft.Extensions.Configuration;
75
using Microsoft.Extensions.Logging;
86
using System;
@@ -68,10 +66,10 @@ public async Task ProcessMessage(IMessage message, CancellationToken token)
6866

6967
using (message.Channel.EnterTypingState())
7068
{
71-
var chatClient = _chatClientFactory.CreateClient($"urn:ollama:{_configuration["OLLAMA_MODEL"]}");
69+
var chatClient = _chatClientFactory.CreateClient("urn:ollama:llama3.2:1b");
7270
var response = await chatClient.GetResponseAsync(new[]
7371
{
74-
new ChatMessage(ChatRole.System, "You are a prompt generator for Stable Diffusion. You will be given a prompt, and you must add details to it to make it better. You must only output the improved prompt, and nothing else."),
72+
new ChatMessage(ChatRole.System, "You are a prompt generator for Stable Diffusion. You will be given a prompt, and you must add a few keywords to it to make it better. You must only output the improved prompt, and nothing else. Do not use command line arguments or negative tags."),
7573
new ChatMessage(ChatRole.User, prompt)
7674
}, new ChatOptions { AdditionalProperties = new AdditionalPropertiesDictionary { { "Think", false } } }, cancellationToken: token);
7775

0 commit comments

Comments
 (0)