Update seed2_tokenizer.py
Browse files- seed2_tokenizer.py +1 -6
seed2_tokenizer.py
CHANGED
|
@@ -29,7 +29,6 @@ import os
|
|
| 29 |
# from timm.models import create_model
|
| 30 |
from typing import Any, Dict, List, Optional, Union
|
| 31 |
from transformers import LlamaTokenizer
|
| 32 |
-
from diffusers import DiffusionPipeline
|
| 33 |
# from torchvision.transforms.functional import pil_to_tensor
|
| 34 |
|
| 35 |
# import torch
|
|
@@ -39,11 +38,7 @@ from torchvision import transforms
|
|
| 39 |
import torch.utils.checkpoint as checkpoint
|
| 40 |
|
| 41 |
DIFFUSION_NAME = 'stabilityai/stable-diffusion-2-1-unclip'
|
| 42 |
-
|
| 43 |
-
# from qformer.qformer_quantizer import Blip2QformerQuantizer
|
| 44 |
-
# from diffusers import StableUnCLIPImg2ImgPipeline
|
| 45 |
-
|
| 46 |
-
from diffusers import StableUnCLIPImg2ImgPipeline
|
| 47 |
|
| 48 |
import logging
|
| 49 |
|
|
|
|
| 29 |
# from timm.models import create_model
|
| 30 |
from typing import Any, Dict, List, Optional, Union
|
| 31 |
from transformers import LlamaTokenizer
|
|
|
|
| 32 |
# from torchvision.transforms.functional import pil_to_tensor
|
| 33 |
|
| 34 |
# import torch
|
|
|
|
| 38 |
import torch.utils.checkpoint as checkpoint
|
| 39 |
|
| 40 |
DIFFUSION_NAME = 'stabilityai/stable-diffusion-2-1-unclip'
|
| 41 |
+
#from diffusers import StableUnCLIPImg2ImgPipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
|
| 43 |
import logging
|
| 44 |
|