摘要： 目录text encoderunetvae text encoder unet conv_in Conv2d(4, 320, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1)) time_proj Timesteps() time_embedding 阅读全文

摘要： 目录加载fp16模型到显存的消耗单图运算时的消耗unet阶段attention_probs = self.get_attention_scoresvae阶段双图运算时的消耗unet阶段vae阶段单图controller = AttentionStore()时的消耗size of attention_ 阅读全文