Skip to content

Commit e4e172a

Browse files
authored
Merge pull request #211 from microsoft/hjiang/support_bf16
Fix(LLMLingua): support bf16
2 parents 9b357b5 + 20d94fb commit e4e172a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

llmlingua/prompt_compressor.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright (c) 2023 Microsoft
1+
# Copyright (c) 2023-2025 Microsoft
22
# Licensed under The MIT License [see LICENSE for details]
33

44
import bisect
@@ -1335,7 +1335,6 @@ def sync_sentence(sentences, text):
13351335
sentence_ppl = [
13361336
self.get_condition_ppl(sentence, question, condition_in_question)
13371337
.cpu()
1338-
.numpy()
13391338
.item()
13401339
for sentence in context_sentences
13411340
]

0 commit comments

Comments
 (0)