Hugging Face's logo

Spaces:
deedax
/
Change-Your-Style
Runtime error

App Files Community
deedax commited on
Commit
959adab
·
1 Parent(s): eb033aa

Fixed issues with concepts learned embeddings

Browse files
Files changed (4) hide show
  1. learned_embeds/arcane_learned_embeds.bin +0 -0
  2. learned_embeds/sims_learned_embeds.bin +0 -0
  3. learned_embeds/uzumaki_learned_embeds.bin +0 -0
  4. utils.py +3 -3
learned_embeds/arcane_learned_embeds.bin ADDED
Binary file (3.82 kB). View file
 
learned_embeds/sims_learned_embeds.bin ADDED
Binary file (3.82 kB). View file
 
learned_embeds/uzumaki_learned_embeds.bin ADDED
Binary file (3.82 kB). View file
 
utils.py CHANGED
@@ -55,11 +55,11 @@ def latents_to_pil(latents):
55
  def get_style(style):
56
  learned_emebeds_map = {
57
  'Ghibli': ['<ghibli-face>', 'ghibli'],
58
- 'Manga': ['<manga-style>', 'manga'],
59
  'GTA 5': ['<gta5-artwork>', 'gta'],
60
- 'Sims': ['<sims-2-portrait>', 'sims'],
61
  'Kaya Ghost Assasin': ['<kaya-ghost-assasin>', 'kaya'],
62
- 'Uzumaki': ['<uzumaki>', 'uzumaki'],
63
  'Arcane': ['<arcane-style-jv>', 'arcane']
64
  }
65
  return learned_emebeds_map[style]
 
55
  def get_style(style):
56
  learned_emebeds_map = {
57
  'Ghibli': ['<ghibli-face>', 'ghibli'],
58
+ 'Manga': ['<manga>', 'manga'],
59
  'GTA 5': ['<gta5-artwork>', 'gta'],
60
+ 'Sims': ['<sims2-portrait>', 'sims'],
61
  'Kaya Ghost Assasin': ['<kaya-ghost-assasin>', 'kaya'],
62
+ 'Uzumaki': ['<NARUTO>', 'uzumaki'],
63
  'Arcane': ['<arcane-style-jv>', 'arcane']
64
  }
65
  return learned_emebeds_map[style]