acheong08 / Diffusion-ColabUI

Choose your diffusion models and spin up a WebUI on Colab in one click
Apache License 2.0
419 stars 62 forks source link

gdrive never works for me #77

Closed RayZ3R0 closed 1 year ago

RayZ3R0 commented 1 year ago

Whenever I run in gdrive, after everything is done downloading, it quits with ^C. I have tried multiple accounts with VPN in multiple locations but it never works when I run in gdrive. VPN and switching account always usually works without gdrive

acheong08 commented 1 year ago

gdrive might take up more RAM as it is stored in memory before saving all at once. You could share it with me and I'll upload it to huggingface

RayZ3R0 commented 1 year ago

Nono I want to run in gdrive so the generated images are stored in gdrive cuz I bulk generate a lot of images so sometimes the colab disconnects and all the images are lost and I rage

acheong08 commented 1 year ago

Ah. Got it. I thought you were downloading from gdrive

acheong08 commented 1 year ago

Ok. Still seems to be a RAM issue. will check it out rn

RayZ3R0 commented 1 year ago

Alright

acheong08 commented 1 year ago

Check if your gdrive storage is full

RayZ3R0 commented 1 year ago

Nope, it has about 12gb free space

acheong08 commented 1 year ago

I could possibly tweak it so that images are saved to drive automatically rather than running the whole thing from drive. Does that work for your purposes?

RayZ3R0 commented 1 year ago

Yeah that'd be good 😊

acheong08 commented 1 year ago

btw running in gdrive is working without ^C for me. Might also be a result of Google limits

acheong08 commented 1 year ago

I do get an error though:

Traceback (most recent call last):
  File "/content/drive/MyDrive/AI/stable-diffusion-webui/modules/call_queue.py", line 45, in f
    res = list(func(*args, **kwargs))
  File "/content/drive/MyDrive/AI/stable-diffusion-webui/modules/call_queue.py", line 28, in f
    res = func(*args, **kwargs)
  File "/content/drive/MyDrive/AI/stable-diffusion-webui/modules/img2img.py", line 152, in img2img
    processed = process_images(p)
  File "/content/drive/MyDrive/AI/stable-diffusion-webui/modules/processing.py", line 479, in process_images
    res = process_images_inner(p)
  File "/content/drive/MyDrive/AI/stable-diffusion-webui/modules/processing.py", line 549, in process_images_inner
    p.init(p.all_prompts, p.all_seeds, p.all_subseeds)
  File "/content/drive/MyDrive/AI/stable-diffusion-webui/modules/processing.py", line 891, in init
    image = images.flatten(img, opts.img2img_background_color)
  File "/content/drive/MyDrive/AI/stable-diffusion-webui/modules/images.py", line 652, in flatten
    if img.mode == "RGBA":
AttributeError: 'NoneType' object has no attribute 'mode'
RayZ3R0 commented 1 year ago

That's certainly weird, I just started the colab the first time today with run in gdrive selected so I don't get why I'm being limited

acheong08 commented 1 year ago

Anyways, I'll try to implement the save to gdrive feature by today

RayZ3R0 commented 1 year ago

Alright, thanks!

acheong08 commented 1 year ago

I added some code to the dev branch for stable-diffusion-webui. Testing it out

acheong08 commented 1 year ago

It was annoying trying to add this as @automatic1111 didn't implement a feature to allow specifying this via CLI. Took some time to get a feel for how the code was organized

acheong08 commented 1 year ago

https://github.com/acheong08/Diffusion-ColabUI/releases/tag/3.0.0

acheong08 commented 1 year ago

Complete and tested. Try it out.

RayZ3R0 commented 1 year ago

Alright, I think I'll need to do it tomorrow, idk why but every time I start its shutting down with a ^C now, probably on high demand or smth

RayZ3R0 commented 1 year ago

This is weird asf, I just started the colab the first time today and it shut down with ^C