Skip to content

Commit 0e4b2a6

Browse files
authored
test : 테스트용 주석 추가입니다 (#8) (#9)
1 parent 69b7cc9 commit 0e4b2a6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

app/models/kobert_classifier.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,8 @@
88
model = AutoModel.from_pretrained(model_name)
99

1010
# 문장 임베딩 생성 함수
11+
#테스트용 문장
12+
1113
def get_sentence_embedding(text: str):
1214
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
1315
with torch.no_grad():

0 commit comments

Comments
 (0)