File size: 265 Bytes
79eeb88
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
import gradio as gr


def show_error(message: str) -> None:
    if "num_inference_steps != 2" in message:
        gr.Warning(
            "Please set the generation setting inference steps to 2 for SANA sprint model"
        )
    else:
        gr.Warning(message)