Views
No views yet
diffusers 库中 AutoencoderKL 架构训练的无条件 VAE 模型,专门用于多肉植物的图像重建与潜空间特征融合。1from diffusers import AutoencoderKL
2import torch
3
4# 加载模型
5model_id = "HaiPenglai/Succulent-VAE-128"
6vae = AutoencoderKL.from_pretrained(model_id)
7
8# 假设你有一张 128x128 的多肉图片 tensor: x
9# 编码到潜空间
10# posterior = vae.encode(x).latent_dist
11# z = posterior.sample()
12# 从潜空间解码
13# reconstruction = vae.decode(z).sample