move download options to editor

undreamai · Jul 26, 2024 · a909ef8 · a909ef8
1 parent 621ca4c
commit a909ef8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/Editor/LLMEditor.cs b/Editor/LLMEditor.cs
@@ -92,8 +92,8 @@ static void ResetModelOptions()
         {
             List<string> existingOptions = new List<string>();
             foreach (ModelEntry entry in LLMManager.modelEntries) existingOptions.Add(entry.url);
-            modelOptions = new List<string>();
-            modelURLs = new List<string>();
+            modelOptions = new List<string>(){"Download model", "Custom URL"};
+            modelURLs = new List<string>(){null, null};
             foreach ((string name, string url) in LLMUnitySetup.modelOptions)
             {
                 if (url != null && existingOptions.Contains(url)) continue;

diff --git a/Runtime/LLMUnitySetup.cs b/Runtime/LLMUnitySetup.cs
@@ -90,8 +90,6 @@ public class LLMUnitySetup
         /// <summary> Default models for download </summary>
         [HideInInspector] public static readonly (string, string)[] modelOptions = new(string, string)[]
         {
-            ("Download model", null),
-            ("Custom URL", null),
             ("Mistral 7B Instruct v0.2 (medium, best overall)", "https://huggingface.co/TheBloke/Mistral-7B-Instruct-v0.2-GGUF/resolve/main/mistral-7b-instruct-v0.2.Q4_K_M.gguf?download=true"),
             ("OpenHermes 2.5 7B (medium, best for conversation)", "https://huggingface.co/TheBloke/OpenHermes-2.5-Mistral-7B-GGUF/resolve/main/openhermes-2.5-mistral-7b.Q4_K_M.gguf?download=true"),
             ("Phi 3 (small, great)", "https://huggingface.co/microsoft/Phi-3-mini-4k-instruct-gguf/resolve/main/Phi-3-mini-4k-instruct-q4.gguf?download=true"),