|
@@ -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:
|