runshouse commited on
Commit
15ab9c6
1 Parent(s): 862693b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="sd-concepts-library", sort="likes", direction=-1)
17
  models = []
18
 
19
- pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", 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")
 
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")