Skip to content

GitLab

Commit f74a28f6 authored by wooway777's avatar wooway777
Browse files

issue/356 - fixed a type error

parent 60ca4508
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
test/infiniop-test/test_generate/infiniop_test.py
View file @ f74a28f6
......@@ -15,7 +15,7 @@ def np_dtype_to_ggml(tensor_dtype: np.dtype):
return GGMLQuantizationType.F32
elif tensor_dtype == np.float64:
return GGMLQuantizationType.F64
elif tensor_dtype == np.bool:
elif tensor_dtype == np.bool_:
return GGMLQuantizationType.Q8_K
elif tensor_dtype == np.int8:
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