silencer107 commited on
Commit
f0ab4bb
·
verified ·
1 Parent(s): 67f2828

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +1 -1
src/pipeline.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  from PIL.Image import Image
3
- from diffusers import UniPCMultistepScheduler
4
  from sfast.compilers.diffusion_pipeline_compiler import (compile,
5
  CompilationConfig)
6
  from pipelines.models import TextToImageRequest
 
1
  import torch
2
  from PIL.Image import Image
3
+ from diffusers import UniPCMultistepScheduler, AutoencoderTiny
4
  from sfast.compilers.diffusion_pipeline_compiler import (compile,
5
  CompilationConfig)
6
  from pipelines.models import TextToImageRequest