henrisel/DeltaVR3DModelGeneration
1
0
Fork 0
forked from cgvr/DeltaVR
Code Pull Requests Activity

trellis and invokeAI urls configurable in config

Browse Source
This commit is contained in:
henrisel
2026-01-19 18:55:25 +02:00
parent 4292750b65
commit ad6954884b
8 changed files with 83 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Assets/_PROJECT/Scripts/ModeGeneration/InvokeAiClient.cs
View File

@@ -14,9 +14,7 @@ public class InvokeAiClient : MonoBehaviour
{
public static InvokeAiClient Instance { get; private set; }
public string INVOKEAI_BASE_URL;
public string DEFAULT_QUEUE_ID = "default";
public string MODEL_KEY;
public string promptSuffix = ", single object, front and side fully visible, realistic style, plain neutral background, clear details, soft studio lighting, true-to-scale";
private HttpClient httpClient;
@@ -27,7 +25,7 @@ public class InvokeAiClient : MonoBehaviour
{
Timeout = TimeSpan.FromSeconds(120)
};
httpClient.BaseAddress = new Uri(INVOKEAI_BASE_URL);
httpClient.BaseAddress = new Uri(ConfigManager.Instance.Config.invokeAIUrl);
Instance = this;
}
@@ -520,13 +518,14 @@ public class InvokeAiClient : MonoBehaviour
public async Task<byte[]> GenerateImage(string prompt)
{
string modelKey = ConfigManager.Instance.Config.invokeAIModelKey;
string refinedPrompt = prompt + promptSuffix;
JObject args = new JObject()
{
["prompt"] = refinedPrompt,
["width"] = 512,
["height"] = 512,
["model_key"] = MODEL_KEY,
["model_key"] = modelKey,
};
UnityEngine.Debug.Log("Starting image generation...");