Spaces:
Configuration error
Configuration error
Update app.py
Browse files
app.py
CHANGED
@@ -13,10 +13,10 @@ import html
|
|
13 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
14 |
|
15 |
api = HfApi()
|
16 |
-
models_list = api.list_models(author="
|
17 |
models = []
|
18 |
|
19 |
-
pipe = StableDiffusionPipeline.from_pretrained("
|
20 |
|
21 |
def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, token=None):
|
22 |
loaded_learned_embeds = torch.load(learned_embeds_path, map_location="cpu")
|
|
|
13 |
from share_btn import community_icon_html, loading_icon_html, share_js
|
14 |
|
15 |
api = HfApi()
|
16 |
+
models_list = api.list_models(author="magichour", sort="likes", direction=-1)
|
17 |
models = []
|
18 |
|
19 |
+
pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=True, revision="fp16", torch_dtype=torch.float16).to("cuda")
|
20 |
|
21 |
def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, token=None):
|
22 |
loaded_learned_embeds = torch.load(learned_embeds_path, map_location="cpu")
|