AI說書 - 從0開始 - 528 | Stable Diffusion 之 Keras 實現 (隨機初始圖生成)

我想要一天分享一點「LLM從底層堆疊的技術」，並且每篇文章長度控制在三分鐘以內，讓大家不會壓力太大，但是又能夠每天成長一點。

Keras 實現 Stable Diffusion 的隨機初始圖片部分為：

def _get_initial_diffusion_noise(self, batch_size, seed):
	if seed is not None:
		return tf.random.stateless_normal((batch_size, self.img_height // 8, self.img_width // 8, 4), seed = [seed, seed])
	else:
		return tf.random.normal((batch_size, self.img_height // 8, self.img_width // 8, 4))