Save clip models to local directory?

It looks like the clip models are stored in a cache on the local machine which works for most use cases; however if we use google colab it basically redownloads them each time when you reopen the notebook.

Is there a way to save them in a folder and then load them from there? Not sure if this is possible in progrock?

I also have gpu’s that I load stuff into remotely; which means they have to download the clip models over and over. The dd finetune model is in a folder. Can the clip models be loaded from a folder, not cache?

"RN50": "https://openaipublic.azureedge.net/clip/models/afeb0e10f9e5a86da6080e35cf09123aca3b358a0c3e3b6c78a7b63bc04b6762/RN50.pt",

[lowfuel]

Should have it checked in later today, just testing a few other things first

[lowfuel]

Probably should file an issue with them asking to support the download_root option ala CLIP, instead of relying soley on a hardcoded directory.

By the way I am probably going to make changing the directory an option in PRD, rather than the default. While I understand your use case, I’ve already discovered it’s not ideal to have to have copies of all the CLIP models in every directory I have for PRD, rather than all of them sharing one cache directory. I usually have 2 or 3 PRD directories I use for various reasons (testing new stuff, testing the current version, testing old versions), plus several other tools I’ve written that also use CLIP, and now I’m having to copy the models folder to every one of them. I know others who use PRD as well as Disco and other versions of Disco on their local machine who appreciate the shared models, too.