Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Downgrade revert torch version after removing optimum-qanto, and other minor version-related fixes.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-23 18:24:58 +00:00 committed by Brandon
parent dff4a88baa
commit 3339a4baf0
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/backend/util/hotfixes.py
View File

@ -3,7 +3,7 @@ from typing import Any, Dict, List, Optional, Tuple, Union
import diffusers import diffusers
import torch import torch
from diffusers.configuration_utils import ConfigMixin, register_to_config from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.loaders.single_file_model import FromOriginalModelMixin from diffusers.loaders import FromOriginalControlNetMixin
from diffusers.models.attention_processor import AttentionProcessor, AttnProcessor from diffusers.models.attention_processor import AttentionProcessor, AttnProcessor
from diffusers.models.controlnet import ControlNetConditioningEmbedding, ControlNetOutput, zero_module from diffusers.models.controlnet import ControlNetConditioningEmbedding, ControlNetOutput, zero_module
from diffusers.models.embeddings import ( from diffusers.models.embeddings import (
@ -32,7 +32,7 @@ from invokeai.backend.util.logging import InvokeAILogger
logger = InvokeAILogger.get_logger(__name__) logger = InvokeAILogger.get_logger(__name__)
class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalModelMixin): class ControlNetModel(ModelMixin, ConfigMixin, FromOriginalControlNetMixin):
""" """
A ControlNet model. A ControlNet model.

4
pyproject.toml
View File

@ -52,10 +52,10 @@ dependencies = [
"sentencepiece==0.2.0", "sentencepiece==0.2.0",
"spandrel==0.3.4", "spandrel==0.3.4",
"timm==0.6.13", # needed to override timm latest in controlnet_aux, see https://github.com/isl-org/ZoeDepth/issues/26 "timm==0.6.13", # needed to override timm latest in controlnet_aux, see https://github.com/isl-org/ZoeDepth/issues/26
"torch==2.4.0", "torch==2.2.2",
"torchmetrics==0.11.4", "torchmetrics==0.11.4",
"torchsde==0.2.6", "torchsde==0.2.6",
"torchvision==0.19.0", "torchvision==0.17.2",
"transformers==4.41.1", "transformers==4.41.1",
# Core application dependencies, pinned for reproducible builds. # Core application dependencies, pinned for reproducible builds.