Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "0cfbb51b0c36f541697e5ef83296bda874ac0671"
Commit c4e9034b authored by rusty1s's avatar rusty1s
Browse files

update to torch nightly

parent cb6c8b71
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/utils.py
View file @ c4e9034b
...@@ -6,7 +6,7 @@ dtypes = get_all_dtypes() ...@@ -6,7 +6,7 @@ dtypes = get_all_dtypes()
dtypes.remove(torch.half) dtypes.remove(torch.half)
dtypes.remove(torch.short) # PyTorch scatter does not work on short types. dtypes.remove(torch.short) # PyTorch scatter does not work on short types.
dtypes.remove(torch.bool) dtypes.remove(torch.bool)
if torch.bfloat16 in dtypes: if hasattr(torch, 'bfloat16'):
dtypes.remove(torch.bfloat16) dtypes.remove(torch.bfloat16)
grad_dtypes = [torch.float, torch.double] grad_dtypes = [torch.float, torch.double]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment