Skip to content

Commit 90f004b

Browse files
authoredFeb 4, 2024
remove benchmarkwrapper form deepspeed example (#10079)
1 parent f9a468a commit 90f004b

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed
 

‎python/llm/example/CPU/Deepspeed-AutoTP/deepspeed_autotp.py

-2
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,6 @@
5050
import intel_extension_for_pytorch as ipex
5151
import time
5252
import argparse
53-
from benchmark_util import BenchmarkWrapper
5453

5554
if __name__ == '__main__':
5655
parser = argparse.ArgumentParser(description='Predict Tokens using `generate()` API for Llama2 model')
@@ -94,7 +93,6 @@
9493
model = model.to(f'cpu:{local_rank}')
9594

9695
print(model)
97-
model = BenchmarkWrapper(model, do_print=True)
9896

9997
# Load tokenizer
10098
tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True)

0 commit comments

Comments
 (0)
Please sign in to comment.