Ruff formatting

This commit is contained in:
Billy
2025-03-21 17:35:12 +11:00
parent fb2c7b9566
commit 3938736bd8
2 changed files with 4 additions and 7 deletions

View File

@@ -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]:

View File

@@ -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