Unverified Commit 28254c79 authored by Anh71me's avatar Anh71me Committed by GitHub
Browse files

Fix type annotation (#5146)

* Fix type annotation on Scheduler.from_pretrained

* Fix type annotation on PIL.Image
parent 0bc6be69
...@@ -3,7 +3,7 @@ import inspect ...@@ -3,7 +3,7 @@ import inspect
from typing import Optional, Union from typing import Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from torch.nn import functional as F from torch.nn import functional as F
from torchvision import transforms from torchvision import transforms
......
...@@ -2,7 +2,7 @@ import inspect ...@@ -2,7 +2,7 @@ import inspect
from typing import List, Optional, Union from typing import List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from torch import nn from torch import nn
from torch.nn import functional as F from torch.nn import functional as F
......
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
from typing import List, Optional, Tuple, Union from typing import List, Optional, Tuple, Union
import PIL import PIL.Image
import torch import torch
from torchvision import transforms from torchvision import transforms
......
...@@ -7,7 +7,7 @@ import warnings ...@@ -7,7 +7,7 @@ import warnings
from typing import List, Optional, Union from typing import List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
from accelerate import Accelerator from accelerate import Accelerator
......
...@@ -2,7 +2,7 @@ import inspect ...@@ -2,7 +2,7 @@ import inspect
from typing import Callable, List, Optional, Tuple, Union from typing import Callable, List, Optional, Tuple, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
......
...@@ -3,7 +3,7 @@ import re ...@@ -3,7 +3,7 @@ import re
from typing import Any, Callable, Dict, List, Optional, Union from typing import Any, Callable, Dict, List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from packaging import version from packaging import version
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
......
...@@ -3,7 +3,7 @@ import re ...@@ -3,7 +3,7 @@ import re
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from packaging import version from packaging import version
from transformers import CLIPImageProcessor, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTokenizer
......
from typing import Any, Callable, Dict, List, Optional, Union from typing import Any, Callable, Dict, List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from diffusers import StableDiffusionImg2ImgPipeline from diffusers import StableDiffusionImg2ImgPipeline
......
...@@ -6,7 +6,7 @@ from typing import Any, Callable, Dict, List, Optional, Union ...@@ -6,7 +6,7 @@ from typing import Any, Callable, Dict, List, Optional, Union
import kornia import kornia
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from packaging import version from packaging import version
from transformers import CLIPFeatureExtractor, CLIPVisionModelWithProjection from transformers import CLIPFeatureExtractor, CLIPVisionModelWithProjection
......
...@@ -16,7 +16,7 @@ import inspect ...@@ -16,7 +16,7 @@ import inspect
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from packaging import version from packaging import version
from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
......
...@@ -24,7 +24,7 @@ from typing import List, Optional, Union ...@@ -24,7 +24,7 @@ from typing import List, Optional, Union
import numpy as np import numpy as np
import onnx import onnx
import onnx_graphsurgeon as gs import onnx_graphsurgeon as gs
import PIL import PIL.Image
import tensorrt as trt import tensorrt as trt
import torch import torch
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
......
...@@ -24,7 +24,7 @@ from typing import List, Optional, Union ...@@ -24,7 +24,7 @@ from typing import List, Optional, Union
import numpy as np import numpy as np
import onnx import onnx
import onnx_graphsurgeon as gs import onnx_graphsurgeon as gs
import PIL import PIL.Image
import tensorrt as trt import tensorrt as trt
import torch import torch
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
......
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
import PIL import PIL.Image
import torch import torch
from transformers import ( from transformers import (
CLIPImageProcessor, CLIPImageProcessor,
......
...@@ -16,7 +16,7 @@ import math ...@@ -16,7 +16,7 @@ import math
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from PIL import Image from PIL import Image
from transformers import CLIPTextModel, CLIPTokenizer from transformers import CLIPTextModel, CLIPTokenizer
......
import inspect import inspect
from typing import List, Optional, Union from typing import List, Optional, Union
import PIL import PIL.Image
import torch import torch
from torch.nn import functional as F from torch.nn import functional as F
from transformers import ( from transformers import (
......
...@@ -16,7 +16,7 @@ import warnings ...@@ -16,7 +16,7 @@ import warnings
from typing import List, Optional, Union from typing import List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from PIL import Image from PIL import Image
......
...@@ -16,7 +16,7 @@ import inspect ...@@ -16,7 +16,7 @@ import inspect
from typing import Any, Callable, Dict, List, Optional, Union from typing import Any, Callable, Dict, List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from packaging import version from packaging import version
from transformers import CLIPImageProcessor, XLMRobertaTokenizer from transformers import CLIPImageProcessor, XLMRobertaTokenizer
......
...@@ -2,7 +2,7 @@ from dataclasses import dataclass ...@@ -2,7 +2,7 @@ from dataclasses import dataclass
from typing import List, Optional, Union from typing import List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
from ...utils import ( from ...utils import (
BaseOutput, BaseOutput,
......
...@@ -38,7 +38,7 @@ from diffusers.utils import numpy_to_pil ...@@ -38,7 +38,7 @@ from diffusers.utils import numpy_to_pil
if is_vision_available(): if is_vision_available():
import PIL import PIL.Image
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
......
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
# limitations under the License. # limitations under the License.
from typing import List, Optional, Union from typing import List, Optional, Union
import PIL import PIL.Image
import torch import torch
from transformers import CLIPTokenizer from transformers import CLIPTokenizer
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment