Spaces:
Sleeping
Sleeping
Commit
·
18adfb4
1
Parent(s):
7a41953
fixing the path of concept bin file
Browse files
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)
|