huggingface-hub 0.24.3__tar.gz → 0.24.6__tar.gz

{huggingface_hub-0.24.3 → huggingface_hub-0.24.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface_hub
-Version: 0.24.3
+Version: 0.24.6
 Summary: Client library to download and publish models, datasets and other repos on the huggingface.co hub
 Home-page: https://github.com/huggingface/huggingface_hub
 Author: Hugging Face, Inc.

{huggingface_hub-0.24.3 → huggingface_hub-0.24.6}/src/huggingface_hub/__init__.py

@@ -46,7 +46,7 @@ import sys
 from typing import TYPE_CHECKING
 
 
-__version__ = "0.24.3"
+__version__ = "0.24.6"
 
 # Alphabetical order of definitions is ensured in tests
 # WARNING: any comment added in this dictionary definition will be lost when

{huggingface_hub-0.24.3 → huggingface_hub-0.24.6}/src/huggingface_hub/file_download.py

@@ -940,7 +940,7 @@ def _create_symlink(src: str, dst: str, new_blob: bool = False) -> None:
     # Symlinks are not supported => let's move or copy the file.
     if new_blob:
         logger.info(f"Symlink not supported. Moving file from {abs_src} to {abs_dst}")
-        shutil.move(abs_src, abs_dst)
+        shutil.move(abs_src, abs_dst, copy_function=_copy_no_matter_what)
     else:
         logger.info(f"Symlink not supported. Copying file from {abs_src} to {abs_dst}")
         shutil.copyfile(abs_src, abs_dst)
@@ -1951,6 +1951,11 @@ def _chmod_and_move(src: Path, dst: Path) -> None:
         tmp_file.touch()
         cache_dir_mode = Path(tmp_file).stat().st_mode
         os.chmod(str(src), stat.S_IMODE(cache_dir_mode))
+    except OSError as e:
+        logger.warning(
+            f"Could not set the permissions on the file '{src}'. "
+            f"Error: {e}.\nContinuing without setting permissions."
+        )
     finally:
         try:
             tmp_file.unlink()
@@ -1959,7 +1964,21 @@ def _chmod_and_move(src: Path, dst: Path) -> None:
             # See https://github.com/huggingface/huggingface_hub/issues/2359
             pass
 
-    shutil.move(str(src), str(dst))
+    shutil.move(str(src), str(dst), copy_function=_copy_no_matter_what)
+
+
+def _copy_no_matter_what(src: str, dst: str) -> None:
+    """Copy file from src to dst.
+
+    If `shutil.copy2` fails, fallback to `shutil.copyfile`.
+    """
+    try:
+        # Copy file with metadata and permission
+        # Can fail e.g. if dst is an S3 mount
+        shutil.copy2(src, dst)
+    except OSError:
+        # Copy only file content
+        shutil.copyfile(src, dst)
 
 
 def _get_pointer_path(storage_folder: str, revision: str, relative_filename: str) -> str:

{huggingface_hub-0.24.3 → huggingface_hub-0.24.6}/src/huggingface_hub/inference/_common.py

@@ -296,7 +296,7 @@ def _format_text_generation_stream_output(
     if not byte_payload.startswith(b"data:"):
         return None  # empty line
 
-    if byte_payload == b"data: [DONE]":
+    if byte_payload.strip() == b"data: [DONE]":
         raise StopIteration("[DONE] signal received.")
 
     # Decode payload
@@ -344,7 +344,7 @@ def _format_chat_completion_stream_output(
     if not byte_payload.startswith(b"data:"):
         return None  # empty line
 
-    if byte_payload == b"data: [DONE]":
+    if byte_payload.strip() == b"data: [DONE]":
         raise StopIteration("[DONE] signal received.")
 
     # Decode payload
@@ -355,7 +355,7 @@ def _format_chat_completion_stream_output(
 
 async def _async_yield_from(client: "ClientSession", response: "ClientResponse") -> AsyncIterable[bytes]:
     async for byte_payload in response.content:
-        yield byte_payload
+        yield byte_payload.strip()
     await client.close()
 
 

{huggingface_hub-0.24.3 → huggingface_hub-0.24.6}/src/huggingface_hub.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface-hub
-Version: 0.24.3
+Version: 0.24.6
 Summary: Client library to download and publish models, datasets and other repos on the huggingface.co hub
 Home-page: https://github.com/huggingface/huggingface_hub
 Author: Hugging Face, Inc.