Hugging Face's logo

Spaces:
tombetthauser
/
astronaut-horse-concept-loader
Runtime error

App Files Community
4
tombetthauser commited on
Commit
aaff49e
·
1 Parent(s): 0508515

Added control net pipe to concept loader func

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -191,7 +191,7 @@ for model in ahx_model_list:
191
  #if token cannot be loaded, skip it
192
  try:
193
  learned_token = load_learned_embed_in_clip(f"{model_id}/learned_embeds.bin", pipe.text_encoder, pipe.tokenizer, token_name)
194
- # _learned_token_controlnet = load_learned_embed_in_clip(f"{model_id}/learned_embeds.bin", controlnet_pipe.text_encoder, controlnet_pipe.tokenizer, token_name)
195
  except:
196
  continue
197
  model_content["token"] = learned_token
 
191
  #if token cannot be loaded, skip it
192
  try:
193
  learned_token = load_learned_embed_in_clip(f"{model_id}/learned_embeds.bin", pipe.text_encoder, pipe.tokenizer, token_name)
194
+ _learned_token_controlnet = load_learned_embed_in_clip(f"{model_id}/learned_embeds.bin", controlnet_pipe.text_encoder, controlnet_pipe.tokenizer, token_name)
195
  except:
196
  continue
197
  model_content["token"] = learned_token