AjayP13 commited on
Commit
a37cee2
·
verified ·
1 Parent(s): 9b8bf44

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,9 +12,9 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  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_model.to(device)
17
- luar_tokenizer = AutoTokenizer.from_pretrained("rrivera1849/LUAR-MUD", trust_remote_code=True)
18
 
19
 
20
  def get_target_style_embeddings(target_texts_batch):
 
12
  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", revision="51b0d9ecec5336314e02f191dd8ca4acc0652fe1", trust_remote_code=True).half()
16
  luar_model.to(device)
17
+ luar_tokenizer = AutoTokenizer.from_pretrained("rrivera1849/LUAR-MUD", revision="51b0d9ecec5336314e02f191dd8ca4acc0652fe1", trust_remote_code=True)
18
 
19
 
20
  def get_target_style_embeddings(target_texts_batch):