Update src/pipeline.py
Browse files- src/pipeline.py +0 -1
src/pipeline.py
CHANGED
|
@@ -8,7 +8,6 @@ from diffusers import DiffusionPipeline, AutoencoderTiny, FluxTransformer2DModel
|
|
| 8 |
from transformers import T5EncoderModel
|
| 9 |
from huggingface_hub.constants import HF_HUB_CACHE
|
| 10 |
from torchao.quantization import quantize_, int8_weight_only, float8_weight_only
|
| 11 |
-
from first_block_cache.diffusers_adapters import apply_cache_on_pipe
|
| 12 |
from caching import apply_cache_on_pipe
|
| 13 |
from pipelines.models import TextToImageRequest
|
| 14 |
from torch import Generator
|
|
|
|
| 8 |
from transformers import T5EncoderModel
|
| 9 |
from huggingface_hub.constants import HF_HUB_CACHE
|
| 10 |
from torchao.quantization import quantize_, int8_weight_only, float8_weight_only
|
|
|
|
| 11 |
from caching import apply_cache_on_pipe
|
| 12 |
from pipelines.models import TextToImageRequest
|
| 13 |
from torch import Generator
|