nyanko7 commited on
Commit
70a03b7
1 Parent(s): 212cf37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -11,7 +11,7 @@ from diffusers import (
11
  DDIMScheduler,
12
  UNet2DConditionModel,
13
  )
14
- from modules.model_pww import (
15
  CrossAttnProcessor,
16
  StableDiffusionPipeline,
17
  load_lora_attn_procs,
@@ -28,6 +28,7 @@ models = [
28
  ("Anything 4.0", "andite/anything-v4.0"),
29
  ("Open Journey", "prompthero/openjourney"),
30
  ("Basil Mix", "nuigurumi/basil_mix"),
 
31
  ("ACertainModel", "JosephusCheung/ACertainModel"),
32
  ]
33
 
@@ -98,11 +99,7 @@ if torch.cuda.is_available():
98
 
99
 
100
  def get_model_list():
101
- model_available = []
102
- for model in models:
103
- if Path(model[1]).is_dir():
104
- model_available.append(model)
105
- return model_available
106
 
107
  unet_cache = {
108
  base_name: unet
@@ -297,7 +294,7 @@ def detect_text(text, state, width, height):
297
  new_state[item] = {
298
  "map": state[item]["map"],
299
  "weight": state[item]["weight"],
300
- "mask_outsides": state[item]["weight"],
301
  }
302
  else:
303
  new_state[item] = {
 
11
  DDIMScheduler,
12
  UNet2DConditionModel,
13
  )
14
+ from modules.model import (
15
  CrossAttnProcessor,
16
  StableDiffusionPipeline,
17
  load_lora_attn_procs,
 
28
  ("Anything 4.0", "andite/anything-v4.0"),
29
  ("Open Journey", "prompthero/openjourney"),
30
  ("Basil Mix", "nuigurumi/basil_mix"),
31
+ ("Pastal Mix", "andite/pastel-mix")
32
  ("ACertainModel", "JosephusCheung/ACertainModel"),
33
  ]
34
 
 
99
 
100
 
101
  def get_model_list():
102
+ return models
 
 
 
 
103
 
104
  unet_cache = {
105
  base_name: unet
 
294
  new_state[item] = {
295
  "map": state[item]["map"],
296
  "weight": state[item]["weight"],
297
+ "mask_outsides": state[item]["mask_outsides"],
298
  }
299
  else:
300
  new_state[item] = {