Colorization uvim model not working about big_vision

[akolesnikoff]

Hi @muhammad-ahmed-ghani,

To get colorful samples please set the sampling temperature to 1.0 in the last cell of the colab: temperature = jnp.array(1.0). After fixing the temperature also consider running inference multiple times and then selecting the best colorization attempt. The near zero sampling temperature is not good default for the colorization task, we will likely update the colab soon.

Let us know whether it worked for you.

[muhammad-ahmed-ghani]

I changed it but still it adds a slight blueish tint to the images.

[akolesnikoff]

That is a bit puzzling, as I am able to get reasonable colorization. If you share a reproduction I may have a look at some point.

[muhammad-ahmed-ghani]

That is a bit puzzling, as I am able to get reasonable colorization. If you share a reproduction I may have a look at some point.

You can check the code here. I just changed the images directories.

https://colab.research.google.com/drive/1E5R_pla2cPkwVKxu9cCgvCt9PtAA-PCq?usp=sharing

[muhammad-ahmed-ghani]

That is a bit puzzling, as I am able to get reasonable colorization. If you share a reproduction I may have a look at some point.

You can check the code here. I just changed the images directories.

https://colab.research.google.com/drive/1E5R_pla2cPkwVKxu9cCgvCt9PtAA-PCq?usp=sharing

Hi @akolesnikoff, have you find anything on it ?