Jetson Container - model download

I am probably being stupid but I have downloaded the stable-diffusion-webui container. However every time that I start this container it downloads a model from hugging-face. [ I am using a Jetson Orin Nano development kit with SSD]

Is is it possible to tell the container to cache the model?

Hi,

The container will reset to the initial status when re-launch.
If you don’t want to download the model each time, you can build a custom image on the top of the container with the downloaded model.

Thanks.

Hi @martin150, if you start the container with jetson-containers/run.sh it will automatically mount your jetson-containers/data directory under /data inside the container, and by default stable-diffusion-webui is launched with --data=/data/models/stable-diffusion - do you see the models under there?

Dusty,

Under jetson-containers/data there is a models directory which has sub directories ‘hugging-face’ (357Mbyte) and stable-diffusion(4.3Gbyte). Despite this it always seems to go and download it again

Hmm okay thanks, for me it does not re-download the SD model. Does it re-download the entire 4.3GB every time? Maybe it is downloading a smaller model that is not cached?

It reloads 8Gbyte every time. Perhaps I should delete the jetson-containers directory and re-install it? Does just deleting the directory delete the containers or do I have to do something else?

You can try that - deleting your local jetson-containers repo will not delete the containers, those are stored separately by your docker daemon. You can then re-clone jetson-containers repo and see if the SD models re-appear under data/ after you run the SD container again. I’m not sure why it would continually re-download them unless there is an issue with the mounting…

You can also start the container to run bash instead of stable-diffusion-webui, then you can poke around inside the container and get a better idea of what mounted files it sees:

./run.sh $(./autotag stable-diffusion-webui) /bin/bash

Dusty,
I deleted jetson-containers and r-installed it. It seems to be working now with 36.2.0, thank you for your help.

Martin

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.