update template

app/src/base_model_template.py

@@ -21,20 +21,20 @@ brushnet = BrushNetModel.from_pretrained(brushnet_path, torch_dtype=torch_dtype)
 
 
 base_models_list = [
-    {
-        "name": "dreamshaper_8 (Preload)",
-        "local_path": "models/base_model/dreamshaper_8",
-        "pipe": StableDiffusionBrushNetPipeline.from_pretrained(
-            "models/base_model/dreamshaper_8", brushnet=brushnet, torch_dtype=torch_dtype, low_cpu_mem_usage=False
-        ).to(device)
-    },
-    {
-        "name": "epicrealism (Preload)",
-        "local_path": "models/base_model/epicrealism_naturalSinRC1VAE",
-        "pipe": StableDiffusionBrushNetPipeline.from_pretrained(
-            "models/base_model/epicrealism_naturalSinRC1VAE", brushnet=brushnet, torch_dtype=torch_dtype, low_cpu_mem_usage=False
-        ).to(device)
-    },
+    # {
+    #     "name": "dreamshaper_8 (Preload)",
+    #     "local_path": "models/base_model/dreamshaper_8",
+    #     "pipe": StableDiffusionBrushNetPipeline.from_pretrained(
+    #         "models/base_model/dreamshaper_8", brushnet=brushnet, torch_dtype=torch_dtype, low_cpu_mem_usage=False
+    #     ).to(device)
+    # },
+    # {
+    #     "name": "epicrealism (Preload)",
+    #     "local_path": "models/base_model/epicrealism_naturalSinRC1VAE",
+    #     "pipe": StableDiffusionBrushNetPipeline.from_pretrained(
+    #         "models/base_model/epicrealism_naturalSinRC1VAE", brushnet=brushnet, torch_dtype=torch_dtype, low_cpu_mem_usage=False
+    #     ).to(device)
+    # },
     {
         "name": "henmixReal (Preload)",
         "local_path": "models/base_model/henmixReal_v5c",

app/src/vlm_template.py

@@ -28,22 +28,22 @@ vlms_list = [
     #             "llava-hf/llava-v1.6-mistral-7b-hf", torch_dtype=torch_dtype, device_map=device
     #         ).to("cpu"),
     # },
-    {
-        "type": "llava-next",
-        "name": "llama3-llava-next-8b-hf (Preload)",
-        "local_path": "models/vlms/llama3-llava-next-8b-hf",
-        "processor": LlavaNextProcessor.from_pretrained(
-            "models/vlms/llama3-llava-next-8b-hf"
-        ) if os.path.exists("models/vlms/llama3-llava-next-8b-hf") else LlavaNextProcessor.from_pretrained(
-            "llava-hf/llama3-llava-next-8b-hf"
-        ),
-        "model": LlavaNextForConditionalGeneration.from_pretrained(
-            "models/vlms/llama3-llava-next-8b-hf", torch_dtype=torch_dtype, device_map=device
-        ).to("cpu") if os.path.exists("models/vlms/llama3-llava-next-8b-hf") else 
-            LlavaNextForConditionalGeneration.from_pretrained(
-                "llava-hf/llama3-llava-next-8b-hf", torch_dtype=torch_dtype, device_map=device
-            ).to("cpu"),
-    },
+    # {
+    #     "type": "llava-next",
+    #     "name": "llama3-llava-next-8b-hf (Preload)",
+    #     "local_path": "models/vlms/llama3-llava-next-8b-hf",
+    #     "processor": LlavaNextProcessor.from_pretrained(
+    #         "models/vlms/llama3-llava-next-8b-hf"
+    #     ) if os.path.exists("models/vlms/llama3-llava-next-8b-hf") else LlavaNextProcessor.from_pretrained(
+    #         "llava-hf/llama3-llava-next-8b-hf"
+    #     ),
+    #     "model": LlavaNextForConditionalGeneration.from_pretrained(
+    #         "models/vlms/llama3-llava-next-8b-hf", torch_dtype=torch_dtype, device_map=device
+    #     ).to("cpu") if os.path.exists("models/vlms/llama3-llava-next-8b-hf") else 
+    #         LlavaNextForConditionalGeneration.from_pretrained(
+    #             "llava-hf/llama3-llava-next-8b-hf", torch_dtype=torch_dtype, device_map=device
+    #         ).to("cpu"),
+    # },
     # {
     #     "type": "llava-next",
     #     "name": "llava-v1.6-vicuna-13b-hf",