Update all files for DiffusionSat-SR-Texas-256

pipeline_diffusionsat.py

@@ -5,11 +5,16 @@ from the checkpoint folder without importing the project package.
 
 from __future__ import annotations
 
+import inspect
 from typing import Any, Callable, Dict, List, Optional, Union
 
 import torch
 from packaging import version
-from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
+try:
+    from transformers import CLIPFeatureExtractor
+except ImportError:  # transformers>=5.0 drops CLIPFeatureExtractor
+    from transformers import CLIPImageProcessor as CLIPFeatureExtractor
+from transformers import CLIPTextModel, CLIPTokenizer
 
 from diffusers.configuration_utils import FrozenDict
 from diffusers.models import AutoencoderKL
@@ -17,10 +22,10 @@ from diffusers.schedulers import KarrasDiffusionSchedulers
 from diffusers.utils import (
     deprecate,
     logging,
-    randn_tensor,
     replace_example_docstring,
     is_accelerate_available,
 )
+from diffusers.utils.torch_utils import randn_tensor
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline
 from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
 from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
@@ -52,6 +57,16 @@ class DiffusionSatPipeline(DiffusionPipeline):
 
     _optional_components = ["safety_checker", "feature_extractor"]
 
+    @classmethod
+    def _get_signature_types(cls):
+        """
+        Override to skip strict type resolution when loading via diffusers 0.36,
+        which cannot resolve the forward references in these custom modules.
+        """
+        required, optional = DiffusionPipeline._get_signature_keys(cls)
+        keys = list(required) + list(optional)
+        return {key: (inspect.Signature.empty,) for key in keys}
+
     def __init__(
         self,
         vae: AutoencoderKL,
@@ -176,7 +191,6 @@ class DiffusionSatPipeline(DiffusionPipeline):
         return md
 
     @torch.no_grad()
-    @replace_example_docstring(EXAMPLE_DOC_STRING)
     def __call__(
         self,
         prompt: Union[str, List[str]] = None,