AttributeError: 'EmptyManager' object has no attribute 'total_steps'

[aheschl1]

When loading the model with

model = VQModelHF.from_pretrained("BorelTHU/optvq-16x16x8")

an error is thrown, unless the model is in eval mode:

if L.log.total_steps % 1000 == 0:
       ^^^^^^^^^^^^^^^^^
AttributeError: 'EmptyManager' object has no attribute 'total_steps'

After model.encode(x)

[zbr17]

Hello, the error message here originates from the recorder and does not affect the model's computation. You can comment out the code branch if L.log.total_steps % 1000 == 0 to prevent the error from occurring in train mode. We will fix this issue in a future version.