ai-toolkit icon indicating copy to clipboard operation
ai-toolkit copied to clipboard

Based on training my own flux model

Open KoDelioDa opened this issue 1 year ago • 3 comments

when I use run.py to train the model, the model is downloaded automatically. I want to know where this model is downloaded to, and I want to replace it with my own base model for training

KoDelioDa avatar Dec 14 '24 07:12 KoDelioDa

The better way would be tweaking the code to read from your own huggingface repo or local machine.

prp-e avatar Dec 15 '24 13:12 prp-e

'name_or_path' in config.

Just point that to the path of your own model.

By default huggingface-hub downloads to cache directory - typically ~/.cache on Linux. Not sure about Windows. You can use huggingface-hub to specify local_dir and put that in 'name_or_path'.

No need to tweak the code base.

0x1355 avatar Jan 22 '25 09:01 0x1355

You can set the HF_HOME variable to change local dir. Environment variables https://huggingface.co/docs/huggingface_hub/main/en/package_reference/environment_variables

On 01/22/2025 11:00 AM CET 0x1355 @.***> wrote:

'name_or_path' in config.

Just point that to the path of your own model.

By default huggingface-hub downloads to cache directory - typically ~/.cache on Linux. Not sure about Windows. You can use huggingface-hub to specify local_dir and put that in 'name_or_path'.

No need to tweak the code base.

— Reply to this email directly, view it on GitHub https://github.com/ostris/ai-toolkit/issues/236#issuecomment-2606788298, or unsubscribe https://github.com/notifications/unsubscribe-auth/AADFU66YVQZWODFHE2W2JS32L5T27AVCNFSM6AAAAABTTJUYQSVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDMMBWG44DQMRZHA. You are receiving this because you are subscribed to this thread.Message ID: @.***>

jsemrau avatar Jan 22 '25 10:01 jsemrau