Google Colab (free plan) Web-UI breaks if file response is bigger than 2MB

[giy-debug]

The Google Colab https://colab.research.google.com/drive/1Iy-xW9t1-OQWhb0hNxueGij8phCyluOh) doesn't work with a batch count higher than two because requests fail when the file response is bigger than 2MB. This also applies to upscales or anything where the file response is bigger than 2MB.

I don't know if this is a limitation of Colab or Gradio but there seems to be no exception handling, that's why all buttons become unresponsive afterwards and you have to reload the page, losing all your inputs.

Any way to fix this with a config parameter or so?

[Puncia]

I've been having the same problems with colab, but I don't think it's strictly related to the size. It's something else. Sometimes it works, sometimes it doesn't.

[Puncia]

gradio-app/gradio#2260

[Extraltodeus]

Even pro has the same issue but I also don't think that this is related to the size.

Note : you can add this :

from time import sleep
from IPython.display import clear_output 

while True:
  clear_output(wait=True)
  sleep(0.1)

At the end of your colab to be able to visualize the terminal's output and see more easily if anything is still running.
I think that the whole issue is more related to how the UI reacts to the end of a job. Since opening a new tab, starting a new job and interrupting it can unlock your first tab. Sometimes it also takes a bit of time and waiting a bit more in between each new batch helps a lot!

[Puncia]

You don't need to loop, you can just add "--gradio-debug" to sys.argv when you launch the webui and the cell will stay looping, while displaying the output of SD

[C43H66N12O12S2]

upstream

[giy-debug]

I found a simple workaround as long as Gradio is not working properly.

Using the latest Colab from https://colab.research.google.com/drive/1kw3egmSn-KgWsikYvOMjJkVDsPLjEMzl

In cell 5 add the installation of ngrok

%cd /content/stable-diffusion-webui
!pip install pyngrok
!git pull

In cell 6

%cd /content/stable-diffusion-webui

ngrok_token = "XXXXXXXXX" #@param {type:"string"}
ngrok_auth = "me:qwerty" #@param {type:"string"}

!ngrok authtoken {ngrok_token}
!nohup ngrok http 7860 -auth={ngrok_auth} -log=stdout >ngrok.log &
!sleep 2
!cat ngrok.log

!COMMANDLINE_ARGS="--listen" REQS_FILE="requirements.txt" python launch.py

you need an account from https://ngrok.com/ and the auth token (see ngrok_token)