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
|