mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-01-09 23:18:04 -05:00
don't skip gguf test if ggml is not installed (#14086)
* don't skip gguf test if ggml is not installed should just let it fail * fix
This commit is contained in:
@@ -69,7 +69,7 @@ testing_minimal = [
|
||||
"hypothesis>=6.148.9",
|
||||
"z3-solver",
|
||||
]
|
||||
testing_unit = ["tinygrad[testing_minimal]", "tqdm", "safetensors", "tabulate", "openai"]
|
||||
testing_unit = ["tinygrad[testing_minimal]", "tqdm", "safetensors", "tabulate", "openai", "ggml-python"]
|
||||
testing = [
|
||||
"tinygrad[testing_unit]",
|
||||
"pillow",
|
||||
@@ -87,7 +87,6 @@ testing = [
|
||||
"networkx",
|
||||
"nibabel",
|
||||
"bottle",
|
||||
"ggml-python",
|
||||
"capstone",
|
||||
"pycocotools",
|
||||
"boto3",
|
||||
|
||||
@@ -1,12 +1,9 @@
|
||||
import os, unittest, ctypes
|
||||
from tinygrad import dtypes, Tensor, fetch, Device
|
||||
import numpy as np
|
||||
from tinygrad.nn.state import ggml_data_to_tensor, gguf_load
|
||||
from tinygrad.device import is_dtype_supported
|
||||
try:
|
||||
import ggml
|
||||
except ModuleNotFoundError:
|
||||
raise unittest.SkipTest("ggml not installed, skipping gguf test")
|
||||
import numpy as np
|
||||
import ggml
|
||||
|
||||
ggml_test_block_count = 4
|
||||
ggml_type_to_np_dtype = {
|
||||
|
||||
Reference in New Issue
Block a user