fix(gui): add strings for more stable diffusion models, nvidia GPUs

ssube · Jan 8, 2023 · 33fd5f1 · 33fd5f1
1 parent 93e53f6
commit 33fd5f1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/gui/src/strings.ts b/gui/src/strings.ts
@@ -1,11 +1,15 @@
 // TODO: set up i18next
 export const MODEL_LABELS = {
+  'stable-diffusion-onnx-v1-4': 'Stable Diffusion v1.4',
   'stable-diffusion-onnx-v1-5': 'Stable Diffusion v1.5',
+  'stable-diffusion-onnx-v2-0': 'Stable Diffusion v2.0',
+  'stable-diffusion-onnx-v2-1': 'Stable Diffusion v2.1',
 };
 
 export const PLATFORM_LABELS: Record<string, string> = {
   amd: 'AMD GPU',
   cpu: 'CPU',
+  nvidia: 'Nvidia GPU',
 };
 
 export const SCHEDULER_LABELS: Record<string, string> = {