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