Spaces:
Build error
Build error
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)
|