Open DarkVamprism opened 1 year ago
Agreed, somewhere like this should work:
extensions/stable-diffusion-webui-rembg/checkpoints
Or even better being able to set it's location with a configuration.
3rd-ed, after all the saving that SD does on your OS Drive/SSD I do not want anymore saving onto it.
Change the location of the rembg models.
extensions/stable-diffusion-webui-rembg/models
File: postprocessing_rembg.py
def process(self, pp: scripts_postprocessing.PostprocessedImage, model, return_mask, alpha_matting, alpha_matting_foreground_threshold, alpha_matting_background_threshold, alpha_matting_erode_size):
if model == "None":
return
import os
script_dir = os.path.abspath( os.path.dirname( __file__ ) )
root_dir = os.path.dirname(os.path.normpath(script_dir))
models_dir = os.path.join(root_dir, "models")
os.environ["U2NET_HOME"] = models_dir
# print("U2NET_HOME: " + models_dir)
.....
Will this problem be corrected?
I've opened a Pull Request to fix this issue: https://github.com/AUTOMATIC1111/stable-diffusion-webui-rembg/pull/28
I believe these should be saved somewhere in the web-ui folders