Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

add support for cache_dir #560

Closed
katossky opened this issue Jun 9, 2023 · 1 comment · Fixed by #561
Closed

add support for cache_dir #560

katossky opened this issue Jun 9, 2023 · 1 comment · Fixed by #561

Comments

@katossky
Copy link

katossky commented Jun 9, 2023

It is not currently possible (as far as I know) to specify where to locate the cache for Peft models.

Regular transformer models now have a cache_dir attributes.

Could it possible to use the same mechanism?
@younesbelkada
Copy link
Contributor

Hi @katossky
After #561 it will be possible to pass cache_dir attribute to PeftModel as the keyword arguments get directly passed to hf_hub_download:

peft/src/peft/peft_model.py

Line 419 in 189a6b8

filename = hf_hub_download(

@younesbelkada younesbelkada mentioned this issue Jun 9, 2023
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

[core] Fix config kwargs younesbelkada/peft
2 participants
@katossky @younesbelkada