add tester for local runs
Browse files
tester.py
ADDED
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import pathlib
|
2 |
+
|
3 |
+
import torch
|
4 |
+
from transformers import pipeline
|
5 |
+
|
6 |
+
def getText():
|
7 |
+
s = '''
|
8 |
+
A US climber has died on his way to scale Mount Everest on Monday, according to an expedition organizer.
|
9 |
+
|
10 |
+
“Jonathan Sugarman died at Camp 2 after he began to feel unwell,” Pasang Sherpa told CNN on Tuesday.
|
11 |
+
|
12 |
+
Seattle-based Sugarman was part of an expedition arranged by Washington state-based International Mountain Guides (IMG) with Beyul Adventure handling the local logistics.
|
13 |
+
|
14 |
+
Sherpa added that “his body remains at Camp 2 with the rest of the climbing team.”
|
15 |
+
|
16 |
+
This comes after Nepal has issued permits for a record 463 climbers by April 26, for this spring season’s expeditions to Mount Everest.
|
17 |
+
|
18 |
+
Following Sugarman’s death, the Embassy of the United States issued a statement. “We can confirm Dr. Jonathan Sugarman passed away while climbing Mt. Everest Monday May 1,” it said. “Our deepest sympathies go out to his family and friends.
|
19 |
+
|
20 |
+
“The Embassy is in contact with Dr. Sugarman’s family and with local authorities. Out of respect for the family’s privacy, we cannot comment further,” read a statement sent to CNN by an Embassy spokesperson.
|
21 |
+
'''
|
22 |
+
|
23 |
+
return s
|
24 |
+
|
25 |
+
def main():
|
26 |
+
print("In main...")
|
27 |
+
# print(pathlib.Path().resolve())
|
28 |
+
# generate_text_pipline = pipeline(model="databricks/dolly-v2-3b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
|
29 |
+
# generate_text_pipline = pipeline(model="verseAI/databricks-dolly-v2-3b", torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
|
30 |
+
workingDir = pathlib.Path().resolve()
|
31 |
+
generate_text_pipline = pipeline(model=workingDir, torch_dtype=torch.bfloat16, trust_remote_code=True, device_map="auto")
|
32 |
+
inputText = getText()
|
33 |
+
resp = generate_text_pipline(inputText)
|
34 |
+
respStr = resp[0]["generated_text"]
|
35 |
+
|
36 |
+
print(f'Input: {inputText}')
|
37 |
+
print(f'Response: {respStr}')
|
38 |
+
|
39 |
+
print("All Done!")
|
40 |
+
|
41 |
+
if __name__ == "__main__":
|
42 |
+
main()
|