Sync the flashrefocus package

.gitignore

@@ -0,0 +1,3 @@
+weights/
+__pycache__/
+*.pyc

flashrefocus/_compat.py

@@ -1,6 +1,9 @@
 """Compatibility with newer libraries than the ones the code was written for."""
 import huggingface_hub
 
-# diffusers 0.25 (required by PiSA-SR's model code) imports huggingface_hub.cached_download, removed in hub 0.26
-if not hasattr(huggingface_hub, "cached_download"):
-    huggingface_hub.cached_download = huggingface_hub.hf_hub_download
+
+def patch_huggingface_hub():
+    """diffusers 0.25 (required by PiSA-SR's model code) imports huggingface_hub.cached_download, removed in hub 0.26.
+    Call before importing diffusers."""
+    if not hasattr(huggingface_hub, "cached_download"):
+        huggingface_hub.cached_download = huggingface_hub.hf_hub_download

flashrefocus/deblurring/decoder_refiner.py

@@ -11,10 +11,12 @@ import torch.nn.functional as F
 from torch import nn
 from torchvision.transforms.functional import gaussian_blur
 
-from .. import _compat  # noqa: F401
 from .. import paths
+from .._compat import patch_huggingface_hub
 from ..layers import NAFBlock
 
+patch_huggingface_hub()   # before diffusers is imported
+
 
 def high_frequency_residual(image, kernel_size=5, sigma=1.0):
     """image minus its blurred 2x down-up-sampled version (one Laplacian-pyramid level)."""

flashrefocus/deblurring/one_step_diffusion.py

@@ -17,8 +17,10 @@ import copy
 
 import torch
 
-from .. import _compat  # noqa: F401
 from .. import paths
+from .._compat import patch_huggingface_hub
+
+patch_huggingface_hub()   # before diffusers is imported
 
 NO_TILING = 100000   # tile sizes above any input: the whole image is processed at once