Spaces:
Running
on
A10G
Running
on
A10G
Commit
·
b78235c
1
Parent(s):
f5c2e01
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,6 @@ from torch import autocast
|
|
| 9 |
from diffusers import StableDiffusionPipeline
|
| 10 |
from huggingface_hub import HfApi
|
| 11 |
from transformers import CLIPTextModel, CLIPTokenizer
|
| 12 |
-
from tqdm.notebook import tqdm
|
| 13 |
|
| 14 |
api = HfApi()
|
| 15 |
models_list = api.list_models(author="sd-concepts-library")
|
|
@@ -48,7 +47,7 @@ def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, tok
|
|
| 48 |
return token
|
| 49 |
|
| 50 |
print("Setting up the public library")
|
| 51 |
-
for model in
|
| 52 |
model_content = {}
|
| 53 |
model_id = model.modelId
|
| 54 |
model_content["id"] = model_id
|
|
|
|
| 9 |
from diffusers import StableDiffusionPipeline
|
| 10 |
from huggingface_hub import HfApi
|
| 11 |
from transformers import CLIPTextModel, CLIPTokenizer
|
|
|
|
| 12 |
|
| 13 |
api = HfApi()
|
| 14 |
models_list = api.list_models(author="sd-concepts-library")
|
|
|
|
| 47 |
return token
|
| 48 |
|
| 49 |
print("Setting up the public library")
|
| 50 |
+
for model in models_list:
|
| 51 |
model_content = {}
|
| 52 |
model_id = model.modelId
|
| 53 |
model_content["id"] = model_id
|