search

philz1337x / clarity-upscaler

Clarity AI | AI Image Upscaler & Enhancer - free and open-source Magnific Alternative
https://ClarityAI.co
GNU Affero General Public License v3.0
3.93k stars 409 forks source link

error when using clarity ai with cog #30

Closed SAKSoomro closed 3 months ago

SAKSoomro commented 6 months ago

I downloaded all 3 models and put into models directory Kindly explain the error,

Starting Docker image cog-clarity-upscaler-base and running setup()... Traceback (most recent call last): File "/root/.pyenv/versions/3.10.4/lib/python3.10/site-packages/cog/server/worker.py", line 189, in _setup run_setup(self._predictor) File "/root/.pyenv/versions/3.10.4/lib/python3.10/site-packages/cog/predictor.py", line 84, in run_setup predictor.setup() File "/src/predict.py", line 33, in setup initialize.imports() File "/src/modules/initialize.py", line 24, in imports from modules import paths, timer, import_hook, errors # noqa: F401 File "/src/modules/paths.py", line 34, in assert sd_path is not None, f"Couldn't find Stable Diffusion in any of: {possible_sd_paths}" AssertionError: Couldn't find Stable Diffusion in any of: ['/src/repositories/stable-diffusion-stability-ai', '.', '/'] ⅹ Model setup failed

datnt-hgmeida commented 6 months ago

you run python download_weights.py to fix this

e-cal commented 6 months ago

also need to add the vae to download_weights.py

# VAE
download_file(
    "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors?download=true",
    "models/VAE",
    "vae-ft-mse-840000-ema-pruned.safetensors"
)
philz1337x commented 3 months ago

@e-cal added a great download-weights.py file to make all this easier!