replicate.exceptions.ModelError: name 'latent_dim' is not defined

[charmdew]

Hello, thank you for your awesome work!

Instead of the pretrained model, I tried to convert the image by uploading a style image with replicate API.
But an error has occured.

Traceback (most recent call last):
  File "C:\\Users\\SSAFY\\PycharmProjects\\JoJoGAN\\jojogan_api.py", line 45, in <module>
    output = version.predict(**inputs)
  File "C:\\Users\\SSAFY\\anaconda3\\envs\\reboot_JoJoGAN\\lib\\site-packages\\replicate\\version.py", line 31, in predict
    raise ModelError(prediction.error)
replicate.exceptions.ModelError: name 'latent_dim' is not defined

---

The same error occurs on the demo page.

[chernikovalexey]

Have you solved it?