only key alt and alt_m9

app.py

@@ -31,13 +31,13 @@ maximum_concepts = 3
 
 #Pre download the files
 if(is_gpu_associated):
-    model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
-    model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
-    model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
+    # model_v1 = snapshot_download(repo_id="multimodalart/sd-fine-tunable")
+    # model_v2 = snapshot_download(repo_id="stabilityai/stable-diffusion-2")
+    # model_v2_512 = snapshot_download(repo_id="stabilityai/stable-diffusion-2-base")
     model_alt = snapshot_download(repo_id="BAAI/AltDiffusion")
     model_alt_m9 = snapshot_download(repo_id="BAAI/AltDiffusion-m9")
     safety_checker = snapshot_download(repo_id="multimodalart/sd-sc")
-    model_to_load = model_v1
+    model_to_load = model_alt_m9
 
 with zipfile.ZipFile("mix.zip", 'r') as zip_ref:
     zip_ref.extractall(".")
@@ -67,16 +67,18 @@ def swap_text(option, base):
 def swap_base_model(selected_model):
     if(is_gpu_associated):
         global model_to_load
-        if(selected_model == "v1-5"):
-            model_to_load = model_v1
-        elif(selected_model == "v2-768"):
-            model_to_load = model_v2
-        elif(selected_model == "alt"):
+        # if(selected_model == "v1-5"):
+        #     model_to_load = model_v1
+        # elif(selected_model == "v2-768"):
+        #     model_to_load = model_v2
+        # elif(selected_model == "alt"):
+        #     model_to_load = model_alt
+        # elif(selected_model == "alt_m9"):
+        #     model_to_load = model_alt_m9
+        # else:
+        #     model_to_load = model_v2_512
+        if(selected_model == "alt"):
             model_to_load = model_alt
-        elif(selected_model == "alt_m9"):
-            model_to_load = model_alt_m9
-        else:
-            model_to_load = model_v2_512
 
 def count_files(*inputs):
     file_counter = 0