satyanayak commited on
Commit
18adfb4
·
1 Parent(s): 7a41953

fixing the path of concept bin file

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,11 +45,11 @@ def generate_images(prompt):
45
  for concept in concepts:
46
  # Load concept embedding
47
  token = load_learned_embed_in_clip(
48
- f"{concept}/learned_embeds.bin",
49
  pipe.text_encoder,
50
  pipe.tokenizer
51
  )
52
-
53
  # Generate random seed
54
  seed = random.randint(1, 999999)
55
  generator = torch.Generator(device=device).manual_seed(seed)
 
45
  for concept in concepts:
46
  # Load concept embedding
47
  token = load_learned_embed_in_clip(
48
+ f"{concept}/blob/main/learned_embeds.bin",
49
  pipe.text_encoder,
50
  pipe.tokenizer
51
  )
52
+
53
  # Generate random seed
54
  seed = random.randint(1, 999999)
55
  generator = torch.Generator(device=device).manual_seed(seed)