Colab broken

[nateraw]

Started investigating fixing it manually in colab with the following (but a PR with these changes + others needed be much appreciated!). The GPU colab in the readme should always work (idc if the TPU one doesn't, as its harder to maintain).

• import issue in app.py: needs to be from stable_diffusion_videos.image_generation import generate_images
• import issue in notebook itself, need to do:

from stable_diffusion_videos.stable_diffusion_pipeline import StableDiffusionWalkPipeline
from stable_diffusion_videos.app import Interface

• jaxlib throws some issue in gpu colab even when we don't touch jax related stuff, due to some import changing and the hardcoded diffusers verison not being compatible. I had to:

pip install jax==0.4.19 jaxlib==0.4.19

• realesrgan has issue on import same as this one. I would guess its fixed upstream and we need to pin new version (hoping it doesn't break something else)

[nateraw]

Fixed by #207