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
...@@ -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 transformers import CLIPImageProcessor from transformers import CLIPImageProcessor
......
...@@ -21,7 +21,7 @@ from typing import Any, Dict, List, Optional, Union ...@@ -21,7 +21,7 @@ from typing import Any, Dict, List, Optional, Union
import flax import flax
import numpy as np import numpy as np
import PIL import PIL.Image
from flax.core.frozen_dict import FrozenDict from flax.core.frozen_dict import FrozenDict
from huggingface_hub import create_repo, snapshot_download from huggingface_hub import create_repo, snapshot_download
from PIL import Image from PIL import Image
......
...@@ -26,7 +26,7 @@ from pathlib import Path ...@@ -26,7 +26,7 @@ from pathlib import Path
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 huggingface_hub import ModelCard, create_repo, hf_hub_download, model_info, snapshot_download from huggingface_hub import ModelCard, create_repo, hf_hub_download, model_info, snapshot_download
from packaging import version from packaging import version
......
...@@ -16,7 +16,7 @@ ...@@ -16,7 +16,7 @@
from typing import List, Optional, Tuple, Union from typing import List, Optional, Tuple, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from ...models import UNet2DModel from ...models import UNet2DModel
......
...@@ -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 BaseOutput from ...utils import BaseOutput
......
...@@ -17,7 +17,7 @@ from dataclasses import dataclass ...@@ -17,7 +17,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
import torch import torch
from transformers import CLIPTextModelWithProjection, CLIPTokenizer from transformers import CLIPTextModelWithProjection, CLIPTokenizer
......
...@@ -16,7 +16,7 @@ from dataclasses import dataclass ...@@ -16,7 +16,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
import torch import torch
from transformers import CLIPImageProcessor, CLIPVisionModel from transformers import CLIPImageProcessor, CLIPVisionModel
......
...@@ -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, CLIPTextModel, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
......
...@@ -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 transformers import CLIPImageProcessor, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTokenizer
......
...@@ -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 transformers import CLIPImageProcessor, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTokenizer
......
...@@ -2,7 +2,7 @@ import inspect ...@@ -2,7 +2,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 transformers import CLIPImageProcessor, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTokenizer
......
...@@ -16,7 +16,7 @@ import inspect ...@@ -16,7 +16,7 @@ import inspect
from typing import Any, Callable, List, Optional, Union from typing import Any, Callable, List, Optional, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from transformers import CLIPImageProcessor, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTokenizer
......
...@@ -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 BaseOutput, is_flax_available from ...utils import BaseOutput, is_flax_available
......
...@@ -17,7 +17,7 @@ import inspect ...@@ -17,7 +17,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 CLIPTextModel, CLIPTokenizer, DPTFeatureExtractor, DPTForDepthEstimation from transformers import CLIPTextModel, CLIPTokenizer, DPTFeatureExtractor, DPTForDepthEstimation
......
...@@ -17,7 +17,7 @@ from dataclasses import dataclass ...@@ -17,7 +17,7 @@ from dataclasses import dataclass
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
......
...@@ -16,7 +16,7 @@ import inspect ...@@ -16,7 +16,7 @@ import inspect
import warnings import warnings
from typing import Any, Callable, Dict, List, Optional, Union from typing import Any, Callable, Dict, List, Optional, Union
import PIL import PIL.Image
import torch import torch
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
......
...@@ -16,7 +16,7 @@ import inspect ...@@ -16,7 +16,7 @@ import inspect
import warnings import warnings
from typing import Any, Callable, Dict, List, Optional, Union from typing import Any, Callable, Dict, List, Optional, Union
import PIL import PIL.Image
import torch import torch
from transformers import ( from transformers import (
CLIPFeatureExtractor, CLIPFeatureExtractor,
......
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
import inspect import inspect
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
import PIL import PIL.Image
import torch import torch
from packaging import version from packaging import version
from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection from transformers import CLIPImageProcessor, CLIPVisionModelWithProjection
......
...@@ -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, CLIPTextModel, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
......
...@@ -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, CLIPTextModel, CLIPTokenizer from transformers import CLIPImageProcessor, CLIPTextModel, 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