update logging and comments

Author: danhipke

src/diffusers/loaders/single_file.py

@@ -311,8 +311,8 @@ def from_single_file(cls, pretrained_model_link_or_path, **kwargs):
                     - A path to a *directory* (for example `./my_pipeline_directory/`) containing the pipeline
                       component configs in Diffusers format.
             no_mmap ('bool', *optional*, defaults to 'False'):
-                Whether to disable mmap when loading the model. This option can perform better when the model is on
-                a network mount or hard drive, which may not handle the seeky-ness of mmap very well.
+                Whether to disable mmap when loading a Safetensors model. This option can perform better when the model
+                is on a network mount or hard drive, which may not handle the seeky-ness of mmap very well.
             kwargs (remaining dictionary of keyword arguments, *optional*):
                 Can be used to overwrite load and saveable variables (the pipeline components of the specific pipeline
                 class). The overwritten components are passed directly to the pipelines `__init__` method. See example

src/diffusers/loaders/single_file_model.py

@@ -171,8 +171,8 @@ def from_single_file(cls, pretrained_model_link_or_path_or_dict: Optional[str] =
                 The specific model version to use. It can be a branch name, a tag name, a commit id, or any identifier
                 allowed by Git.
             no_mmap ('bool', *optional*, defaults to 'False'):
-                Whether to disable mmap when loading the model. This option can perform better when the model is on
-                a network mount or hard drive, which may not handle the seeky-ness of mmap very well.
+                Whether to disable mmap when loading a Safetensors model. This option can perform better when the model
+                is on a network mount or hard drive, which may not handle the seeky-ness of mmap very well.
             kwargs (remaining dictionary of keyword arguments, *optional*):
                 Can be used to overwrite load and saveable variables (for example the pipeline components of the
                 specific pipeline class). The overwritten components are directly passed to the pipelines `__init__`

src/diffusers/models/model_loading_utils.py

@@ -139,10 +139,8 @@ def load_state_dict(checkpoint_file: Union[str, os.PathLike], variant: Optional[
         file_extension = os.path.basename(checkpoint_file).split(".")[-1]
         if file_extension == SAFETENSORS_FILE_EXTENSION:
             if no_mmap:
-                logger.warning("fast load: " + str(checkpoint_file))
                 return safetensors.torch.load(open(checkpoint_file, "rb").read())
             else:
-                logger.warning("slow load: " + str(checkpoint_file))
                 return safetensors.torch.load_file(checkpoint_file, device="cpu")
         else:
             weights_only_kwarg = {"weights_only": True} if is_torch_version(">=", "1.13") else {}