实用程序
用于处理 🤗 Diffusers 的实用程序和辅助函数。
numpy_to_pil
将 NumPy 图像或一批图像转换为 PIL 图像。
pt_to_pil
将 Torch 图像转换为 PIL 图像。
load_image
diffusers.utils.load_image
< 源代码 >( image: Union convert_method: Optional = None ) → PIL.Image.Image
将image
加载到 PIL Image 中。
export_to_gif
export_to_video
diffusers.utils.export_to_video
< 源代码 > ( video_frames: Union output_video_path: str = None fps: int = 10 )
make_image_grid
准备一个图像网格。用于可视化目的。
randn_tensor
diffusers.utils.torch_utils.randn_tensor
< 源代码 > ( shape: Union generator: Union = None device: Optional = None dtype: Optional = None layout: Optional = None )
一个用于在所需的device
上创建具有所需dtype
的随机张量的辅助函数。当传递生成器列表时,可以单独为每个批次大小设置种子。如果传递了 CPU 生成器,则张量始终在 CPU 上创建。