Skip to content

GitLab

Unverified Commit 0f250ec4 authored by PanZezhong1725's avatar PanZezhong1725 Committed by GitHub
Browse files

Merge pull request #357 from InfiniTensor/issue/356 

issue/356 - fixed a type error
parents f8e9d924 f74a28f6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/infiniop-test/test_generate/infiniop_test.py
View file @ 0f250ec4
...@@ -15,7 +15,7 @@ def np_dtype_to_ggml(tensor_dtype: np.dtype): ...@@ -15,7 +15,7 @@ def np_dtype_to_ggml(tensor_dtype: np.dtype):
return GGMLQuantizationType.F32 return GGMLQuantizationType.F32
elif tensor_dtype == np.float64: elif tensor_dtype == np.float64:
return GGMLQuantizationType.F64 return GGMLQuantizationType.F64
elif tensor_dtype == np.bool: elif tensor_dtype == np.bool_:
return GGMLQuantizationType.Q8_K return GGMLQuantizationType.Q8_K
elif tensor_dtype == np.int8: elif tensor_dtype == np.int8:
return GGMLQuantizationType.I8 return GGMLQuantizationType.I8
......
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