i

2026-08-16 13:13:16 +02:00 · 2023-08-09 14:00:59 -07:00
parent 573dc98eff
commit 0b72417588
2 changed files with 2 additions and 2 deletions
@@ -118,7 +118,7 @@ torch.cuda.empty_cache()
 torch.cuda.ipc_collect()

 shape = (1, 4, 128, 128)
-randn = torch.randn(shape).cuda()
+randn = torch.randn(shape).to(torch.float16).cuda()


 def denoiser(input, sigma, c):