Nota-NetsPresso / BK-SDM

A Compressed Stable Diffusion for Efficient Text-to-Image Generation [ECCV'24]
Other
238 stars 16 forks source link

Queries #56

Closed abhigoku10 closed 2 months ago

abhigoku10 commented 5 months ago

@bokyeong1015 hi thanks for sharing this wonderful work , i had few queries and request

  1. Can you please share ur checkpoint-45000 on one drive or google drive , i wanted to test it on the system as donot have the resources to train it on gpu system
  2. In ur paper u have mentioned deploying on Nvidia orin ? did u test it on any other platforms like Nvidia Agx / Nx/ nano if so whats the time taken on it
  3. When deploying it on nvidia orin did u used docker or straight up with hugging face models
  4. The Techniques used in this paper and the snap fusion can we bring in those in this code and can we expect to see some better improvements
  5. image

Thanks n advanc

deepkyu commented 3 months ago

Sorry for late late reply @abhigoku10

For question 3, we utilized stable diffusion webui code and a Docker environment on NVIDIA Jetson, provided by dusty-nv/jetson-containers. We updated the model code to support both the original stable diffusion and our BK-SDM. The model checkpoint is the same as the one available on Hugging Face, but it requires remapping the module name.

bokyeong1015 commented 3 months ago

Hi,