We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 877508f commit fb1f693Copy full SHA for fb1f693
swift/llm/utils/argument.py
@@ -119,7 +119,7 @@ def select_dtype(
119
raise ValueError(f'args.dtype: {self.dtype}')
120
# cuda, npu
121
if self.dtype == 'AUTO':
122
- if is_torch_bf16_gpu_available():
+ if not is_torch_bf16_gpu_available():
123
self.dtype = 'fp16'
124
else:
125
model_torch_dtype = MODEL_MAPPING[self.model_type].get(
0 commit comments