Skip to content

Commit 3a42c8e

Browse files
jainapurvafacebook-github-bot
authored andcommitted
Skip tests on fbcode
Summary: Skip tests on fbcode. Missing model checkpoints Differential Revision: D67982501
1 parent 4738377 commit 3a42c8e

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

test/quantization/test_gptq_mt.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,12 @@
88
from torchao._models.llama.tokenizer import get_tokenizer
99
from torchao.quantization.GPTQ_MT import Int4WeightOnlyGPTQQuantizer, MultiTensor
1010
from torchao.quantization.utils import _lm_eval_available
11+
from torchao.utils import is_fbcode
12+
13+
if is_fbcode():
14+
pytest.skip(
15+
"Skipping the test in fbcode since we don't have TARGET file for kernels"
16+
)
1117

1218
if _lm_eval_available:
1319
hqq_core = pytest.importorskip("hqq.core", reason="requires hqq")

0 commit comments

Comments
 (0)