1

Author: zhanghuiyao

examples/transformers/gpt_oss/generate.py

@@ -11,7 +11,7 @@
 
 def generate(args):
 
-    pipe = pipeline("text-generation", model=args.model_name, mindspore_dtype=mindspore.bfloat16, attn_implementation="eager")
+    pipe = pipeline("text-generation", model=args.model_name, mindspore_dtype=mindspore.bfloat16, model_kwargs={'attn_implementation': 'eager'})
 
     messages = [
         {"role": "user", "content": args.prompt},