Skip to content
This repository was archived by the owner on Sep 10, 2025. It is now read-only.

Commit fd0f53e

Browse files
authored
Update build_utils.py
typo
1 parent ae69428 commit fd0f53e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchchat/utils/build_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def name_to_dtype(name, device):
177177
return torch.float16
178178

179179
# if it's not CUDA, we know it's bfloat16
180-
if "cuda" is not in device_str:
180+
if "cuda" not in device_str:
181181
return torch.bfloat16
182182

183183
if major >= 9:

0 commit comments

Comments
 (0)