Skip to content

assert freeze_codebook if self.freeze_codebook in init #209

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.

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

Closed
wants to merge 1 commit into from

Conversation

rqi3
Copy link

@rqi3 rqi3 commented Apr 11, 2025

Currently, the freeze_codebook as an init argument in the VectorQuantize class is not used. It should be either be used to enforce a check that freeze_codebook=True when self.freeze_codebook=True, or removed as an argument to init.

@lucidrains
Copy link
Owner

@rqi3 oh good catch, thank you

so I think it is probably better to have it be overridable on forwards

let me get that in there within an hour (in transit atm)

lucidrains added a commit that referenced this pull request Apr 11, 2025
@lucidrains
Copy link
Owner

@rqi3 could you see if the latest commit works for you?

@rqi3
Copy link
Author

rqi3 commented Apr 11, 2025

Looks good!

@rqi3 rqi3 closed this Apr 11, 2025
@rqi3 rqi3 deleted the patch-1 branch April 11, 2025 21:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants