moyanwang commited on
Commit
c2fa725
1 Parent(s): b8227c1

add pytorch vae model

Browse files
lyraSD/__pycache__/__init__.cpython-38.pyc ADDED
Binary file (185 Bytes). View file
 
lyraSD/__pycache__/inference.cpython-38.pyc ADDED
Binary file (2.07 kB). View file
 
lyraSD/muse_trt/__pycache__/__init__.cpython-38.pyc ADDED
Binary file (501 Bytes). View file
 
lyraSD/muse_trt/__pycache__/models.cpython-38.pyc ADDED
Binary file (23.9 kB). View file
 
lyraSD/muse_trt/__pycache__/sd_img2img.cpython-38.pyc ADDED
Binary file (9.84 kB). View file
 
lyraSD/muse_trt/__pycache__/sd_text2img.cpython-38.pyc ADDED
Binary file (7.7 kB). View file
 
lyraSD/muse_trt/__pycache__/super.cpython-38.pyc ADDED
Binary file (1.96 kB). View file
 
lyraSD/muse_trt/__pycache__/utilities.cpython-38.pyc ADDED
Binary file (16.3 kB). View file
 
lyraSD/muse_trt/sd_img2img.py CHANGED
@@ -68,7 +68,7 @@ class TRTStableDiffusionImg2ImgPipeline:
68
  self.trt_torch_models_cls = {
69
  'clip': CLIP(),
70
  'unet_fp16': UNet(),
71
- 'vae-encoder': VAEEncoder(),
72
  'vae-decoder': VAEDecoder()
73
  }
74
 
 
68
  self.trt_torch_models_cls = {
69
  'clip': CLIP(),
70
  'unet_fp16': UNet(),
71
+ #'vae-encoder': VAEEncoder(),
72
  'vae-decoder': VAEDecoder()
73
  }
74
 
output/sd-img2img-0.jpg CHANGED
output/sd-text2img-0.jpg CHANGED
sd1.5-engine/{vae-encoder.plan → vae/diffusion_pytorch_model.bin} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9c264f6511de7f69b39c678a4c76afa324ec30b49b7df2cf3ec11d4db329e943
3
- size 69481674
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4302e1efa25f3a47ceb7536bc335715ad9d1f203e90c2d25507600d74006e89
3
+ size 334715313