mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-14 16:22:28 +00:00
Fixed typo in comment
This commit is contained in:
@@ -30,7 +30,7 @@ def copy_stochastic_(target: torch.Tensor, source: torch.Tensor):
|
||||
|
||||
# Kahan summation for bfloat16
|
||||
# The implementation was provided by araleza.
|
||||
# Base on paper "Revisiting BFloat16 Training": https://arxiv.org/pdf/2010.06192
|
||||
# Based on paper "Revisiting BFloat16 Training": https://arxiv.org/pdf/2010.06192
|
||||
|
||||
kahan_residuals = []
|
||||
tensor_index = 0
|
||||
|
||||
Reference in New Issue
Block a user