manbeast3b commited on
Commit
b6f39a3
·
verified ·
1 Parent(s): 092e238

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -19,7 +19,7 @@ from torchao.quantization import quantize_, int8_weight_only, float8_weight_only
19
  from first_block_cache.diffusers_adapters import apply_cache_on_pipe
20
  from pipelines.models import TextToImageRequest
21
  from torch import Generator
22
- from utils import _load
23
 
24
  # Configuration
25
  @dataclass
 
19
  from first_block_cache.diffusers_adapters import apply_cache_on_pipe
20
  from pipelines.models import TextToImageRequest
21
  from torch import Generator
22
+ from para_attn.first_block_cache.diffusers_adapters import apply_cache_on_pipe
23
 
24
  # Configuration
25
  @dataclass