Browse Source

Update chat_completion.py

Kautilya Kondragunta 1 year ago
parent
commit
b4c71bebee
1 changed files with 1 additions and 1 deletions
  1. 1 1
      examples/chat_completion/chat_completion.py

+ 1 - 1
examples/chat_completion/chat_completion.py

@@ -34,7 +34,7 @@ def main(
     enable_azure_content_safety: bool=False, # Enable safety check with Azure content safety api
     enable_sensitive_topics: bool=False, # Enable check for sensitive topics using AuditNLG APIs
     enable_saleforce_content_safety: bool=True, # Enable safety check woth Saleforce safety flan t5
-    use_fast_kernels: bool = False, # Enable using SDPA from PyTroch Accelerated Transformers, make use Flash Attention and Xformer memory-efficient kernels
+    use_fast_kernels: bool = False, # Enable using SDPA from PyTorch Accelerated Transformers, make use Flash Attention and Xformer memory-efficient kernels
     **kwargs
 ):
     if prompt_file is not None: