search

Stability-AI / StableCascade

Official Code for Stable Cascade
MIT License
6.44k stars 518 forks source link

ModuleNotFoundError: No module named 'torchtools.transforms' #77

Closed ZhangYang-UCAS closed 4 months ago

ZhangYang-UCAS commented 4 months ago

ModuleNotFoundError Traceback (most recent call last) Cell In[1], line 10 8 from inference.utils import * 9 from core.utils import load_or_fail ---> 10 from train import ControlNetCore, WurstCoreB 12 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") 13 print(device)

File /StableCascade/train/init.py:1 ----> 1 from .train_b import WurstCore as WurstCoreB 2 from .train_c import WurstCore as WurstCoreC 3 from .train_c_controlnet import WurstCore as ControlNetCore

File /StableCascade/train/train_b.py:14 12 from gdf import GDF, EpsilonTarget, CosineSchedule 13 from gdf import VPScaler, CosineTNoiseCond, DDPMSampler, P2LossWeight, AdaptiveLossWeight ---> 14 from torchtools.transforms import SmartCrop 16 from modules.effnet import EfficientNetEncoder 17 from modules.stage_a import StageA

ModuleNotFoundError: No module named 'torchtools.transforms'

ggsddu7 commented 4 months ago

pip install git+https://github.com/pabloppp/pytorch-tools