Tonic commited on
Commit
e382347
·
1 Parent(s): 757241b

fix typo in kernels community

Browse files
Files changed (1) hide show
  1. app_alternative.py +1 -1
app_alternative.py CHANGED
@@ -12,7 +12,7 @@ try:
12
  "openai/gpt-oss-20b",
13
  torch_dtype="auto",
14
  device_map="auto",
15
- attn_implementation="kernel-community/vllm-flash-attention3"
16
  )
17
  tokenizer = AutoTokenizer.from_pretrained("openai/gpt-oss-20b")
18
 
 
12
  "openai/gpt-oss-20b",
13
  torch_dtype="auto",
14
  device_map="auto",
15
+ attn_implementation="kernels-community/vllm-flash-attention3"
16
  )
17
  tokenizer = AutoTokenizer.from_pretrained("openai/gpt-oss-20b")
18