app/core/custom_warnings.py

import warnings

def custom_warning_filter(message, category, filename, lineno, file=None, line=None):
    if "Torch was not compiled with flash attention" in str(message):
        return  # Suppress this specific warning
    warnings.defaultaction(message, category, filename, lineno, file, line)

def custom_warning():
    warnings.showwarning = custom_warning_filter  # Apply the filter