Bug: [Remote] VAEs don't load

[JohnDotGoat]

This time only 1 of my VAES does work, for the rest of them it throws the next message

File "/content/sd-inference-server/server.py", line 112, in run self.wrapper.txt2img() File "/usr/local/lib/python3.10/dist-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(*args, *kwargs) File "/content/sd-inference-server/wrapper.py", line 611, in txt2img self.load_models(initial_networks) File "/content/sd-inference-server/wrapper.py", line 274, in load_models self.vae = self.storage.get_vae(self.vae_name, self.device) File "/content/sd-inference-server/storage.py", line 313, in get_vae return self.get_component(name, "VAE", device) File "/content/sd-inference-server/storage.py", line 269, in get_component self.file_cache[file] = self.load_file(file, comp) File "/content/sd-inference-server/storage.py", line 378, in load_file state_dict, metadata = convert.convert(file) File "/content/sd-inference-server/convert.py", line 390, in convert return convert_checkpoint(model_path) File "/content/sd-inference-server/convert.py", line 276, in convert_checkpoint state_dict = torch.load(in_file, map_location="cpu") File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 809, in load return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args) File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1172, in _load result = unpickler.load() File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1165, in find_class return super().find_class(mod_name, name) File "/content/sd-inference-server/venv/lib/python3.10/site-packages/pytorch_lightning/init.py", line 34, in from pytorch_lightning.callbacks import Callback # noqa: E402 File "/content/sd-inference-server/venv/lib/python3.10/site-packages/pytorch_lightning/callbacks/init.py", line 25, in from pytorch_lightning.callbacks.progress import ProgressBarBase, RichProgressBar, TQDMProgressBar File "/content/sd-inference-server/venv/lib/python3.10/site-packages/pytorch_lightning/callbacks/progress/init.py", line 22, in from pytorch_lightning.callbacks.progress.rich_progress import RichProgressBar # noqa: F401 File "/content/sd-inference-server/venv/lib/python3.10/site-packages/pytorch_lightning/callbacks/progress/rich_progress.py", line 20, in from torchmetrics.utilities.imports import _compare_version ImportError: cannot import name '_compare_version' from 'torchmetrics.utilities.imports' (/usr/local/lib/python3.10/dist-packages/torchmetrics/utilities/imports.py)

[arenasys]

should be fixed by https://github.com/arenasys/sd-inference-server/commit/f09138c40c4433c82f211ce4a2aa1f946afffc55

[arenasys]

unfixed, the strict unpickler was too strict, had to remove it for now

[arenasys]

aight