Spaces:
Running
Running
Update app.py
Browse files
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):
|