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: