nyanko7 commited on
Commit
a18a27e
1 Parent(s): 1e66485

feat: limit models amount

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -25,14 +25,13 @@ import modules.safe as _
25
  from modules.lora import LoRANetwork
26
 
27
  models = [
28
- # format: name, model_path, clip_skip
29
- ("AbyssOrangeMix2", "Korakoe/AbyssOrangeMix2-HF", 2),
30
- ("Basil Mix", "nuigurumi/basil_mix", 2),
31
- ("Pastal Mix", "andite/pastel-mix", 2),
32
- ("ACertainModel", "JosephusCheung/ACertainModel", 2),
33
  ]
34
 
35
- base_name, base_model, clip_skip = models[0]
 
36
 
37
  samplers_k_diffusion = [
38
  ("Euler a", "sample_euler_ancestral", {}),
@@ -132,8 +131,8 @@ def get_model(name):
132
  return g_unet, g_lora
133
 
134
  # precache on huggingface
135
- # for model in get_model_list():
136
- # get_model(model[0])
137
 
138
  def error_str(error, title="Error"):
139
  return (
 
25
  from modules.lora import LoRANetwork
26
 
27
  models = [
28
+ ("AbyssOrangeMix2", "Korakoe/AbyssOrangeMix2-HF"),
29
+ ("Basil Mix", "nuigurumi/basil_mix"),
30
+ ("Pastal Mix", "andite/pastel-mix"),
 
 
31
  ]
32
 
33
+ base_name, base_model, = models[0]
34
+ clip_skip = 2
35
 
36
  samplers_k_diffusion = [
37
  ("Euler a", "sample_euler_ancestral", {}),
 
131
  return g_unet, g_lora
132
 
133
  # precache on huggingface
134
+ for model in get_model_list():
135
+ get_model(model[0])
136
 
137
  def error_str(error, title="Error"):
138
  return (