[diffusion] chore: clean excessive document (#16986)

This commit is contained in:
Mick
2026-01-13 21:33:50 +08:00
committed by GitHub
parent 888d7e54d1
commit 7a869045b6
27 changed files with 36 additions and 236 deletions

View File

@@ -596,11 +596,7 @@ def set_mixed_precision_policy(
def get_compute_dtype() -> torch.dtype:
"""Get the current compute dtype from mixed precision policy.
Returns:
torch.dtype: The compute dtype to use, defaults to get_default_dtype() if no policy set
"""
"""Get the current compute dtype from mixed precision policy."""
if not hasattr(_mixed_precision_state, "state"):
return torch.get_default_dtype()
else: