Skip to content

Commit 6383e6c

Browse files
jerryzh168facebook-github-bot
authored andcommitted
Fix internal tests after recent chagnes (#2726)
Summary: att Reviewed By: shreydesai, andrewor14 Differential Revision: D79936256
1 parent ec9961c commit 6383e6c

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

test/integration/test_loading_deprecated_checkpoint.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
)
1515
from transformers import AutoModelForCausalLM, AutoTokenizer, TorchAoConfig
1616

17-
from torchao.utils import is_sm_at_least_89
17+
from torchao.utils import is_fbcode, is_sm_at_least_89
1818

1919
_MODEL_NAME_AND_VERSIONS = [
2020
("torchao-testing/opt-125m-float8dq-row-v1-0.13-dev", 1),
@@ -23,6 +23,10 @@
2323

2424
@unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
2525
@unittest.skipIf(not is_sm_at_least_89(), "Nedd sm89+")
26+
@unittest.skipIf(
27+
is_fbcode(),
28+
"Skipping the test in fbcode for now, not sure how to download from transformers",
29+
)
2630
class TestLoadingDeprecatedCheckpoint(TestCase):
2731
@common_utils.parametrize("model_name_and_version", _MODEL_NAME_AND_VERSIONS)
2832
def test_load_model_and_run(self, model_name_and_version):

0 commit comments

Comments
 (0)