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 @@ ...@@ -16,7 +16,7 @@
import unittest import unittest
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from diffusers.image_processor import VaeImageProcessor from diffusers.image_processor import VaeImageProcessor
......
...@@ -26,7 +26,7 @@ import unittest ...@@ -26,7 +26,7 @@ import unittest
import unittest.mock as mock import unittest.mock as mock
import numpy as np import numpy as np
import PIL import PIL.Image
import requests_mock import requests_mock
import safetensors.torch import safetensors.torch
import torch import torch
......
...@@ -11,7 +11,7 @@ import uuid ...@@ -11,7 +11,7 @@ import uuid
from typing import Callable, Union from typing import Callable, Union
import numpy as np import numpy as np
import PIL import PIL.Image
import torch import torch
from huggingface_hub import delete_repo from huggingface_hub import delete_repo
from transformers import CLIPTextConfig, CLIPTextModel, CLIPTokenizer from transformers import CLIPTextConfig, 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