From 58cc68ada43d9661a0ade05ca7e24ffe999deb86 Mon Sep 17 00:00:00 2001 From: Raphael Glon Date: Mon, 27 May 2024 11:16:19 +0200 Subject: [PATCH] wip Signed-off-by: Raphael Glon --- tests/test_modeling_utils.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/tests/test_modeling_utils.py b/tests/test_modeling_utils.py index 2ed70a757aaf36..2df721d0abc299 100755 --- a/tests/test_modeling_utils.py +++ b/tests/test_modeling_utils.py @@ -229,13 +229,8 @@ def test_offline(self): hub_api = HfApi() local_dir = hub_api.snapshot_download(TINY_IMAGE_CLASSIF, cache_dir=tmpdir) - print("TMPDIR, %s" % tmpdir) - print("LOCAL DIR %s", local_dir) - LOG.info("Model %s downloaded in %s", TINY_IMAGE_CLASSIF, local_dir) - import pdb - pdb.set_trace() AutoModelForImageClassification.from_pretrained(TINY_IMAGE_CLASSIF, revision="main", use_auth_token=None) finally: @@ -273,9 +268,6 @@ def test_local_files_only(self): hub_api = HfApi() local_dir = hub_api.snapshot_download(TINY_IMAGE_CLASSIF, cache_dir=tmpdir) - print("TMPDIR, %s" % tmpdir) - print("LOCAL DIR %s", local_dir) - LOG.info("Model %s downloaded in %s", TINY_IMAGE_CLASSIF, local_dir) AutoModelForImageClassification.from_pretrained(TINY_IMAGE_CLASSIF, revision="main",