Skip to content

Commit f0e5a10

Browse files
William-AnCopilot
andauthored
Update src/cuda/huggingface/helloworld.py
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
1 parent afda9ea commit f0e5a10

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/cuda/huggingface/helloworld.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@ def main():
1313
helloworldParser.add_argument("--max_length", type=int, default=256)
1414
args = helloworldParser.parse_args()
1515

16-
hellowordStr = f"Running Hello world for \"{args.model_name}\" with prompt: \"{args.prompt}\""
17-
print(hellowordStr)
18-
print("=" * len(hellowordStr))
16+
helloworld_str = f"Running Hello world for \"{args.model_name}\" with prompt: \"{args.prompt}\""
17+
print(helloworld_str)
18+
print("=" * len(helloworld_str))
1919

2020
pipe = transformers.pipeline(task="text-generation", model=args.model_name, device=args.device)
2121
results = pipe(args.prompt, max_length=args.max_length, truncation=True)

0 commit comments

Comments
 (0)