Hugging Face's logo

Spaces:
tinystyler
/
tinystyler_demo
Running

App Files Community
AjayP13 commited on
Commit
b6f4412
·
verified ·
1 Parent(s): 0174717

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -13,6 +13,7 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
13
  model.to(device)
14
  embedding_model = SentenceTransformer('AnnaWegmann/Style-Embedding', device='cpu').half()
15
  luar_model = AutoModel.from_pretrained("rrivera1849/LUAR-MUD", trust_remote_code=True).half()
 
16
  luar_tokenizer = AutoTokenizer.from_pretrained("rrivera1849/LUAR-MUD", trust_remote_code=True)
17
 
18
 
 
13
  model.to(device)
14
  embedding_model = SentenceTransformer('AnnaWegmann/Style-Embedding', device='cpu').half()
15
  luar_model = AutoModel.from_pretrained("rrivera1849/LUAR-MUD", trust_remote_code=True).half()
16
+ luar_model.to(device)
17
  luar_tokenizer = AutoTokenizer.from_pretrained("rrivera1849/LUAR-MUD", trust_remote_code=True)
18
 
19