mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-04-23 03:00:31 -04:00
Ruff formatting
This commit is contained in:
@@ -21,9 +21,9 @@ Validation errors will raise an InvalidModelConfigException error.
|
||||
"""
|
||||
|
||||
# pyright: reportIncompatibleVariableOverride=false
|
||||
import json
|
||||
import logging
|
||||
import time
|
||||
import json
|
||||
from abc import ABC, abstractmethod
|
||||
from enum import Enum
|
||||
from inspect import isabstract
|
||||
@@ -392,7 +392,6 @@ class ModelConfigBase(ABC, BaseModel):
|
||||
if "source_type" in overrides:
|
||||
overrides["source_type"] = ModelSourceType(overrides["source_type"])
|
||||
|
||||
|
||||
@classmethod
|
||||
def from_model_on_disk(cls, mod: ModelOnDisk, **overrides):
|
||||
"""Creates an instance of this config or raises InvalidModelConfigException."""
|
||||
@@ -668,6 +667,7 @@ class FluxReduxConfig(LegacyProbeMixin, ModelConfigBase):
|
||||
|
||||
class LlavaOnevisionConfig(DiffusersConfigBase, ModelConfigBase):
|
||||
"""Model config for Llava Onevision models."""
|
||||
|
||||
type: Literal[ModelType.LlavaOnevision] = ModelType.LlavaOnevision
|
||||
format: Literal[ModelFormat.Diffusers] = ModelFormat.Diffusers
|
||||
|
||||
@@ -684,10 +684,7 @@ class LlavaOnevisionConfig(DiffusersConfigBase, ModelConfigBase):
|
||||
return False
|
||||
|
||||
architectures = config.get("architectures")
|
||||
return (
|
||||
architectures and
|
||||
architectures[0] == "LlavaOnevisionForConditionalGeneration"
|
||||
)
|
||||
return architectures and architectures[0] == "LlavaOnevisionForConditionalGeneration"
|
||||
|
||||
@classmethod
|
||||
def parse(cls, mod: ModelOnDisk) -> dict[str, Any]:
|
||||
|
||||
@@ -148,7 +148,7 @@ def test_regression_against_model_probe(datadir: Path, override_model_loading):
|
||||
configs_with_tests = set()
|
||||
model_paths = ModelSearch().search(datadir / "stripped_models")
|
||||
fake_hash = "abcdefgh" # skip hashing to make test quicker
|
||||
fake_key = "123" # fixed uuid for comparison
|
||||
fake_key = "123" # fixed uuid for comparison
|
||||
|
||||
for path in model_paths:
|
||||
legacy_config = new_config = None
|
||||
|
||||
Reference in New Issue
Block a user