Detail and no-detail mode

ml-energy · Sep 20, 2024 · 4e4fca8 · 4e4fca8
1 parent 2eb5843
commit 4e4fca8
Show file tree

Hide file tree

Showing 7 changed files with 383 additions and 200 deletions.
diff --git a/app.py b/app.py
diff --git a/data/diffusion/image-to-video/models.json b/data/diffusion/image-to-video/models.json
@@ -2,22 +2,22 @@
   "ali-vilab/i2vgen-xl": {
     "url": "https://huggingface.co/ali-vilab/i2vgen-xl",
     "nickname": "I2VGen XL",
-    "total_params": "2.5B",
-    "denoising_params": "1.4B",
+    "total_params": 2.5,
+    "denoising_params": 1.4,
     "resolution": "1280x720"
   },
   "stabilityai/stable-video-diffusion-img2vid": {
     "url": "https://huggingface.co/stabilityai/stable-video-diffusion-img2vid",
     "nickname": "Stable Video Diffusion",
-    "total_params": "2.3B",
-    "denoising_params": "1.5B",
+    "total_params": 2.3,
+    "denoising_params": 1.5,
     "resolution": "1024x576"
   },
   "stabilityai/stable-video-diffusion-img2vid-xt": {
     "url": "https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt",
     "nickname": "Stable Video Diffusion xt",
-    "total_params": "2.3B",
-    "denoising_params": "1.5B",
+    "total_params": 2.3,
+    "denoising_params": 1.5,
     "resolution": "1024x576"
   }
 }
diff --git a/data/diffusion/text-to-image/models.json b/data/diffusion/text-to-image/models.json
@@ -2,57 +2,57 @@
   "kandinsky-community/kandinsky-2-2-decoder": {
     "url": "https://huggingface.co/kandinsky-community/kandinsky-2-2-decoder",
     "nickname": "Kandinsky 2.2",
-    "total_params": "4.9B",
-    "denoising_params": "1.3B",
+    "total_params": 4.9,
+    "denoising_params": 1.3,
     "resolution": "512x512"
   },
   "kandinsky-community/kandinsky-3": {
     "url": "https://huggingface.co/kandinsky-community/kandinsky-3",
     "nickname": "Kandinsky 3",
-    "total_params": "12.0B",
-    "denoising_params": "3.1B",
+    "total_params": 12.0,
+    "denoising_params": 3.1,
     "resolution": "1024x1024"
   },
   "prompthero/openjourney-v4": {
     "url": "https://huggingface.co/prompthero/openjourney-v4",
     "nickname": "Openjourney V4",
-    "total_params": "1.1B",
-    "denoising_params": "0.9B",
+    "total_params": 1.1,
+    "denoising_params": 0.9,
     "resolution": "512x512"
   },
   "segmind/SSD-1B": {
     "url": "https://huggingface.co/segmind/SSD-1B",
     "nickname": "SSD 1B",
-    "total_params": "2.2B",
-    "denoising_params": "1.3B",
+    "total_params": 2.2,
+    "denoising_params": 1.3,
     "resolution": "1024x1024"
   },
   "stabilityai/sdxl-turbo": {
     "url": "https://huggingface.co/stabilityai/sdxl-turbo",
     "nickname": "Stable Diffusion XL Turbo",
-    "total_params": "3.5B",
-    "denoising_params": "2.6B",
+    "total_params": 3.5,
+    "denoising_params": 2.6,
     "resolution": "512x512"
   },
   "stabilityai/stable-diffusion-2-1": {
     "url": "https://huggingface.co/stabilityai/stable-diffusion-2-1",
     "nickname": "Stable Diffusion 2.1",
-    "total_params": "1.3B",
-    "denoising_params": "0.9B",
+    "total_params": 1.3,
+    "denoising_params": 0.9,
     "resolution": "768x768"
   },
   "stabilityai/stable-diffusion-3-medium-diffusers": {
     "url": "https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers",
     "nickname": "Stable Diffusion 3 Medium",
-    "total_params": "7.7B",
-    "denoising_params": "2.0B",
+    "total_params": 7.7,
+    "denoising_params": 2.0,
     "resolution": "1024x1024"
   },
   "stabilityai/stable-diffusion-xl-base-1.0": {
     "url": "https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0",
     "nickname": "Stable Diffusion XL Base 1.0",
-    "total_params": "3.5B",
-    "denoising_params": "2.6B",
+    "total_params": 3.5,
+    "denoising_params": 2.6,
     "resolution": "1024x1024"
   }
 }
diff --git a/data/diffusion/text-to-video/models.json b/data/diffusion/text-to-video/models.json
@@ -2,15 +2,15 @@
   "ali-vilab/text-to-video-ms-1.7b": {
     "url": "https://huggingface.co/ali-vilab/text-to-video-ms-1.7b",
     "nickname": "ModelScope T2V",
-    "total_params": "1.8B",
-    "denoising_params": "1.4B",
+    "total_params": 1.8,
+    "denoising_params": 1.4,
     "resolution": "256x256"
   },
   "guoyww/animatediff-motion-adapter-v1-5-3": {
     "url": "https://huggingface.co/guoyww/animatediff-motion-adapter-v1-5-3",
     "nickname": "Animatediff",
-    "total_params": "1.9B",
-    "denoising_params": "1.3B",
+    "total_params": 1.9,
+    "denoising_params": 1.3,
     "resolution": "512x512"
   }
 }
diff --git a/data/llm_text_generation/chat/models.json b/data/llm_text_generation/chat/models.json
@@ -2,71 +2,71 @@
   "google/gemma-2-27b-it": {
     "url": "https://huggingface.co/google/gemma-2-27b-it",
     "nickname": "Gemma 2 27B",
-    "params": "27B"
+    "params": 27
   },
   "google/gemma-2-2b-it": {
     "url": "https://huggingface.co/google/gemma-2-2b-it",
     "nickname": "Gemma 2 2B",
-    "params": "2B"
+    "params": 2
   },
   "google/gemma-2-9b-it": {
     "url": "https://huggingface.co/google/gemma-2-9b-it",
     "nickname": "Gemma 2 9B",
-    "params": "9B"
+    "params": 9
   },
   "meta-llama/Meta-Llama-3.1-70B-Instruct": {
     "url": "https://huggingface.co/meta-llama/Meta-Llama-3.1-70B-Instruct",
     "nickname": "Llama 3.1 70B",
-    "params": "70B"
+    "params": 70
   },
   "meta-llama/Meta-Llama-3.1-405B-Instruct": {
     "url": "https://huggingface.co/meta-llama/Meta-Llama-3.1-405B-Instruct",
     "nickname": "Llama 3.1 405B",
-    "params": "405B"
+    "params": 405
   },
   "meta-llama/Meta-Llama-3.1-8B-Instruct": {
     "url": "https://huggingface.co/meta-llama/Meta-Llama-3.1-8B-Instruct",
     "nickname": "Llama 3.1 8B",
-    "params": "8B"
+    "params": 8
   },
   "microsoft/Phi-3-medium-4k-instruct": {
     "url": "https://huggingface.co/microsoft/Phi-3-medium-4k-instruct",
     "nickname": "Phi 3 Medium",
-    "params": "14B"
+    "params": 14
   },
   "microsoft/Phi-3-mini-4k-instruct": {
     "url": "https://huggingface.co/microsoft/Phi-3-mini-4k-instruct",
     "nickname": "Phi 3 Mini",
-    "params": "4B"
+    "params": 4
   },
   "microsoft/Phi-3-small-8k-instruct": {
     "url": "https://huggingface.co/microsoft/Phi-3-small-8k-instruct",
     "nickname": "Phi 3 Small",
-    "params": "7B"
+    "params": 7
   },
   "mistralai/Mistral-7B-Instruct-v0.3": {
     "url": "https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.3",
     "nickname": "Mistral 7B",
-    "params": "7B"
+    "params": 7
   },
   "mistralai/Mistral-Large-Instruct-2407": {
     "url": "https://huggingface.co/mistralai/Mistral-Large-Instruct-2407",
     "nickname": "Mistral Large",
-    "params": "123B"
+    "params": 123
   },
   "mistralai/Mistral-Nemo-Instruct-2407": {
     "url": "https://huggingface.co/mistralai/Mistral-Nemo-Instruct-2407",
     "nickname": "Mistral Nemo",
-    "params": "12B"
+    "params": 12
   },
   "mistralai/Mixtral-8x22B-Instruct-v0.1": {
     "url": "https://huggingface.co/mistralai/Mixtral-8x22B-Instruct-v0.1",
     "nickname": "Mixtral 8x22B",
-    "params": "141B"
+    "params": 141
   },
   "mistralai/Mixtral-8x7B-Instruct-v0.1": {
     "url": "https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1",
     "nickname": "Mixtral 8x7B",
-    "params": "47B"
+    "params": 47
   }
 }
diff --git a/data/llm_text_generation/code/models.json b/data/llm_text_generation/code/models.json
@@ -2,46 +2,46 @@
   "bigcode/starcoder2-15b": {
     "url": "https://huggingface.co/bigcode/starcoder2-15b",
     "nickname": "Starcoder2 15B",
-    "params": "15B"
+    "params": 15
   },
   "bigcode/starcoder2-3b": {
     "url": "https://huggingface.co/bigcode/starcoder2-3b",
     "nickname": "Starcoder2 3B",
-    "params": "3B"
+    "params": 3
   },
   "bigcode/starcoder2-7b": {
     "url": "https://huggingface.co/bigcode/starcoder2-7b",
     "nickname": "Starcoder2 7B",
-    "params": "7B"
+    "params": 7
   },
   "codellama/CodeLlama-13b-hf": {
     "url": "https://huggingface.co/codellama/CodeLlama-13b-hf",
     "nickname": "CodeLlama 13B",
-    "params": "13B"
+    "params": 13
   },
   "codellama/CodeLlama-34b-hf": {
     "url": "https://huggingface.co/codellama/CodeLlama-34b-hf",
     "nickname": "CodeLlama 34B",
-    "params": "34B"
+    "params": 34
   },
   "codellama/CodeLlama-70b-hf": {
     "url": "https://huggingface.co/codellama/CodeLlama-70b-hf",
     "nickname": "CodeLlama 70B",
-    "params": "70B"
+    "params": 70
   },
   "codellama/CodeLlama-7b-hf": {
     "url": "https://huggingface.co/codellama/CodeLlama-7b-hf",
     "nickname": "CodeLlama 7B",
-    "params": "7B"
+    "params": 7
   },
   "google/codegemma-1.1-2b": {
     "url": "https://huggingface.co/google/codegemma-1.1-2b",
     "nickname": "CodeGemma 2B",
-    "params": "2B"
+    "params": 2
   },
   "google/codegemma-7b": {
     "url": "https://huggingface.co/google/codegemma-7b",
     "nickname": "CodeGemma 7B",
-    "params": "7B"
+    "params": 7
   }
 }
diff --git a/data/mllm_text_generation/chat/models.json b/data/mllm_text_generation/chat/models.json
@@ -2,31 +2,31 @@
   "facebook/chameleon-30b": {
     "url": "https://huggingface.co/facebook/chameleon-30b",
     "nickname": "Chameleon 30B",
-    "params": "34B"
+    "params": 34
   },
   "facebook/chameleon-7b": {
     "url": "https://huggingface.co/facebook/chameleon-7b",
     "nickname": "Chameleon 7B",
-    "params": "7B"
+    "params": 7
   },
   "llava-hf/llama3-llava-next-8b-hf": {
     "url": "https://huggingface.co/llava-hf/llama3-llava-next-8b-hf",
     "nickname": "LLaVA NeXT 8B",
-    "params": "8B"
+    "params": 8
   },
   "llava-hf/llava-1.5-13b-hf": {
     "url": "https://huggingface.co/llava-hf/llava-1.5-13b-hf",
     "nickname": "LLaVA 1.5 13B",
-    "params": "13B"
+    "params": 13
   },
   "llava-hf/llava-1.5-7b-hf": {
     "url": "https://huggingface.co/llava-hf/llava-1.5-7b-hf",
     "nickname": "LLaVA 1.5 7B",
-    "params": "7B"
+    "params": 7
   },
   "microsoft/Phi-3-vision-128k-instruct": {
     "url": "https://huggingface.co/microsoft/Phi-3-vision-128k-instruct",
     "nickname": "Phi 3 Vision",
-    "params": "4B"
+    "params": 4
   }
 }