tombetthauser commited on
Commit
14c84e8
Β·
1 Parent(s): 759a079

Add temp diagnostic print for model list

Browse files
Files changed (1) hide show
  1. app.py +16 -1
app.py CHANGED
@@ -126,6 +126,21 @@ def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, tok
126
 
127
  ahx_model_list = [model for model in models_list if "ahx" in model.modelId]
128
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
129
  print("Setting up the public library")
130
  for model in ahx_model_list:
131
  model_content = {}
@@ -235,7 +250,7 @@ def checkbox_block():
235
  def infer(text):
236
  #with autocast("cuda"):
237
  images_list = pipe(
238
- [text]*2,
239
  num_inference_steps=50,
240
  guidance_scale=7.5
241
  )
 
126
 
127
  ahx_model_list = [model for model in models_list if "ahx" in model.modelId]
128
 
129
+ from time import sleep
130
+ print("--------------------------------------------------")
131
+ print("--------------------------------------------------")
132
+ print("--------------------------------------------------")
133
+ print("models_list:")
134
+ print(models_list)
135
+ print("--------------------------------------------------")
136
+ print("ahx_model_list:")
137
+ print(ahx_model_list)
138
+ print("--------------------------------------------------")
139
+ print("--------------------------------------------------")
140
+ print("--------------------------------------------------")
141
+ sleep(60)
142
+
143
+
144
  print("Setting up the public library")
145
  for model in ahx_model_list:
146
  model_content = {}
 
250
  def infer(text):
251
  #with autocast("cuda"):
252
  images_list = pipe(
253
+ [text],
254
  num_inference_steps=50,
255
  guidance_scale=7.5
256
  )