search

Fanghua-Yu / SUPIR

SUPIR aims at developing Practical Algorithms for Photo-Realistic Image Restoration In the Wild. Our new online demo is also released at suppixel.ai.
http://supir.xpixel.group/
Other
4.42k stars 386 forks source link

Conflict #91

Open S0ulan opened 8 months ago

S0ulan commented 8 months ago

Help

(SUPIR) sl00m@sl00m-H87-D3H:~/Projekte/Deeplearning/SUPIR$ python gradio_demo.py 
╭─────────────────────────────── Traceback (most recent call last) ────────────────────────────────╮
│ /home/sl00m/Projekte/Deeplearning/SUPIR/gradio_demo.py:6 in <module>                             │
│                                                                                                  │
│     3 import gradio as gr                                                                        │
│     4 from gradio_imageslider import ImageSlider                                                 │
│     5 import argparse                                                                            │
│ ❱   6 from SUPIR.util import HWC3, upscale_image, fix_resize, convert_dtype                      │
│     7 import numpy as np                                                                         │
│     8 import torch                                                                               │
│     9 from SUPIR.util import create_SUPIR_model, load_QF_ckpt                                    │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/SUPIR/util.py:8 in <module>                              │
│                                                                                                  │
│     5 from PIL import Image                                                                      │
│     6 from torch.nn.functional import interpolate                                                │
│     7 from omegaconf import OmegaConf                                                            │
│ ❱   8 from sgm.util import instantiate_from_config                                               │
│     9                                                                                            │
│    10                                                                                            │
│    11 def get_state_dict(d):                                                                     │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/sgm/__init__.py:1 in <module>                            │
│                                                                                                  │
│ ❱ 1 from .models import AutoencodingEngine, DiffusionEngine                                      │
│   2 from .util import get_configs_path, instantiate_from_config                                  │
│   3                                                                                              │
│   4 __version__ = "0.1.0"                                                                        │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/sgm/models/__init__.py:1 in <module>                     │
│                                                                                                  │
│ ❱ 1 from .autoencoder import AutoencodingEngine                                                  │
│   2 from .diffusion import DiffusionEngine                                                       │
│   3                                                                                              │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/sgm/models/autoencoder.py:12 in <module>                 │
│                                                                                                  │
│     9 from packaging import version                                                              │
│    10 from safetensors.torch import load_file as load_safetensors                                │
│    11                                                                                            │
│ ❱  12 from ..modules.diffusionmodules.model import Decoder, Encoder                              │
│    13 from ..modules.distributions.distributions import DiagonalGaussianDistribution             │
│    14 from ..modules.ema import LitEma                                                           │
│    15 from ..util import default, get_obj_from_str, instantiate_from_config                      │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/sgm/modules/__init__.py:1 in <module>                    │
│                                                                                                  │
│ ❱ 1 from .encoders.modules import GeneralConditioner                                             │
│   2 from .encoders.modules import GeneralConditionerWithControl                                  │
│   3 from .encoders.modules import PreparedConditioner                                            │
│   4                                                                                              │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/sgm/modules/encoders/modules.py:22 in <module>           │
│                                                                                                  │
│     19 )                                                                                         │
│     20                                                                                           │
│     21 from ...modules.autoencoding.regularizers import DiagonalGaussianRegularizer              │
│ ❱   22 from ...modules.diffusionmodules.model import Encoder                                     │
│     23 from ...modules.diffusionmodules.openaimodel import Timestep                              │
│     24 from ...modules.diffusionmodules.util import extract_into_tensor, make_beta_schedule      │
│     25 from ...modules.distributions.distributions import DiagonalGaussianDistribution           │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/sgm/modules/diffusionmodules/__init__.py:6 in <module>   │
│                                                                                                  │
│   3 from .loss import StandardDiffusionLoss                                                      │
│   4 from .model import Decoder, Encoder, Model                                                   │
│   5 from .openaimodel import UNetModel                                                           │
│ ❱ 6 from .sampling import BaseDiffusionSampler                                                   │
│   7 from .wrappers import OpenAIWrapper                                                          │
│   8                                                                                              │
│                                                                                                  │
│ /home/sl00m/Projekte/Deeplearning/SUPIR/sgm/modules/diffusionmodules/sampling.py:20 in <module>  │
│                                                                                                  │
│    17 │   to_sigma,                                                                              │
│    18 )                                                                                          │
│    19 from ...util import append_dims, default, instantiate_from_config                          │
│ ❱  20 from k_diffusion.sampling import get_sigmas_karras, BrownianTreeNoiseSampler               │
│    21                                                                                            │
│    22 DEFAULT_GUIDER = {"target": "sgm.modules.diffusionmodules.guiders.IdentityGuider"}         │
│    23                                                                                            │
╰──────────────────────────────────────────────────────────────────────────────────────────────────╯
ModuleNotFoundError: No module named 'k_diffusion'
(SUPIR) sl00m@sl00m-H87-D3H:~/Projekte/Deeplearning/SUPIR$
S0ulan commented 8 months ago

Solved with Pinokio. Standalone impossible to make it run.