fix: minor fix

app.py

@@ -35,8 +35,8 @@ alt_models = [
     ("Stable Diffusion V1.5", "runwayml/stable-diffusion-v1-5", 1),
     ("Anything V3.0", "Linaqruf/anything-v3.0", 2),
     ("Open Journey", "prompthero/openjourney", 1),
-    ("Eimis AnimeDiffusion", "eimiss/EimisAnimeDiffusion_1.0v", 2)
-    ("Dreamlike Photoreal 2.0", "dreamlike-art/dreamlike-photoreal-2.0", 1)
+    ("Eimis AnimeDiffusion", "eimiss/EimisAnimeDiffusion_1.0v", 2),
+    ("Dreamlike Photoreal 2.0", "dreamlike-art/dreamlike-photoreal-2.0", 1),
     ("Redshift Diffusion", "nitrosocke/redshift-diffusion", 1)
 ]
 
@@ -115,13 +115,14 @@ lora_cache = {
 }
 
 def get_model(name):
-    keys = [k[0] for k in models]
+    local_models = models + alt_models
+    keys = [k[0] for k in local_models]
     if name not in unet_cache:
         if name not in keys:
             raise ValueError(name)
         else:
             unet = UNet2DConditionModel.from_pretrained(
-                models[keys.index(name)][1],
+                local_models[keys.index(name)][1],
                 subfolder="unet",
                 torch_dtype=torch.float16,
             )
@@ -133,10 +134,11 @@ def get_model(name):
     g_lora = lora_cache[name]
     g_unet.set_attn_processor(CrossAttnProcessor())
     g_lora.reset()
+    clip_skip = local_models[keys.index(name)][2]
     if torch.cuda.is_available():
         g_unet.to("cuda")
         g_lora.to("cuda")
-    return g_unet, g_lora, models[keys.index(name)][2]
+    return g_unet, g_lora, clip_skip
 
 # precache on huggingface
 for model in models: