wip

Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>
huggingface · May 27, 2024 · 58cc68a · 58cc68a
1 parent c3d72e2
commit 58cc68a
Showing 1 changed file with 0 additions and 8 deletions.
diff --git 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",