AttributeError: 'dict' object has no attribute 'unfreeze'

[JJYSTEM]

When training Video Model, using the default command, met the object attribute error:

• Command

python scripts/train_vqgan.py -o viper_rl_data/checkpoints/dmc_vqgan -c viper_rl/configs/vqgan/dmc.yaml

• Error

Traceback (most recent call last):
File "scripts/train_vqgan.py", line 202, in
main()
File "scripts/train_vqgan.py", line 55, in main
state, lr_fn = init_model_state_vqgan(init_rng, model, batch, config)
File "/mnt/nas/jiaojiao/2025/viper_rl/viper_rl/videogpt/train_utils.py", line 140, in init_model_state_vqgan
variables = model.vqgan.init(
AttributeError: 'dict' object has no attribute 'unfreeze'

Here is the error line in script:
https://github.com/Alescontrela/viper_rl/blob/b1c847cb76deb8723cfe4cd2fec9e29661ca7a56/viper_rl/videogpt/train_utils.py#L144

https://github.com/Alescontrela/viper_rl/blob/b1c847cb76deb8723cfe4cd2fec9e29661ca7a56/viper_rl/videogpt/train_utils.py#L152

Any idea are welcome.

[yangzhao-666]

Removing .unfreeze() works for me.