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.

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

Sample fix #3

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Conversation

jesperdlau
Copy link

I encountered some bugs while sampling. Maybe it's just me, but when instantiating the model in
src/walkjump/cmdline/_sample.py, The instantiate_model_for_sample_mode function doesn't seem to accept the DictConfig input. This is solved by either giving the specific inputs like
def instantiate_model_for_sample_mode( model_type, checkpoint_path, denoise_path) -> NoiseEnergyModel | DenoiseModel:

Or by instantiating the model by calling the function directly, like what is done with instantiate_redesign_mask and instantiate_seeds.

I also added some examples and comments.

@justinlubin
Copy link

Hey, I just wanted to say that this was a really helpful PR for me! I could get the model up and running with the changes you suggested. Thanks!

@justinlubin
Copy link

justinlubin commented Apr 16, 2024

I also figured out that you need to specify the following option or something similar (e.g. in src/walkjump/hydra_config/sample.yaml) to enable sampling:

redesign_regions: ["L1", "L2", "H1", "H2"]

Otherwise, every index gets masked, and sampling never changes the seed values!

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.

None yet

2 participants