diffusers/benchmark
0
1o
2 %'e� � @ s� d dl Zd dlZd dlmZ d dlZdZejZe� d�Z ej3dded�Z e � e � er5ej
e jdd d4�e _dd efd
d�Zejdd��%Zejdd dd d d�Ze�d�jdddd�Zejeegd� W d � n1 skw Y ejd d� dS )� N)�AutoPipelineForText2ImageFzcuda:0z(stabilityai/stable-diffusion-xl-base-1.0�fp16)�variant� torch_dtypezreduce-overheadT)�mode� fullgraph� �num_images_per_promptc C sT d}d}t � � }t|| |d�j t � � }td|� d�|| � td|| | � d S )N�Maaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa�( )r �num_inference_stepszFor z stepsz Avg per step)�time�pipeline�images�print)r �promptr �5start_time�end_time� r � /home/patrick/benchmark/app.py�generate s r z style.css)�cssz6Batch size� )�label�minimum�maximum�step�valuez7Benchmark!)FTTF)�margin�rounded�8full_width)�fn�inputs)�share)r )�gradio�gr�torch� diffusersr r
�USE_TORCH_COMPILE�float16�dtype�device�from_pretrainedr �to�compile�unet�intr �Blocks�demo�Slider�9batch_size�Button�style�btn�click�launchr r r r �<module> s8 1011 �12� �