We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent afda9ea commit f0e5a10Copy full SHA for f0e5a10
src/cuda/huggingface/helloworld.py
@@ -13,9 +13,9 @@ def main():
13
helloworldParser.add_argument("--max_length", type=int, default=256)
14
args = helloworldParser.parse_args()
15
16
- hellowordStr = f"Running Hello world for \"{args.model_name}\" with prompt: \"{args.prompt}\""
17
- print(hellowordStr)
18
- print("=" * len(hellowordStr))
+ helloworld_str = f"Running Hello world for \"{args.model_name}\" with prompt: \"{args.prompt}\""
+ print(helloworld_str)
+ print("=" * len(helloworld_str))
19
20
pipe = transformers.pipeline(task="text-generation", model=args.model_name, device=args.device)
21
results = pipe(args.prompt, max_length=args.max_length, truncation=True)
0 commit comments