Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: add method get_scheme to Metadata

Browse Source
This commit is contained in:
Manuel Schmid
2024-02-02 01:19:21 +01:00
parent ea6839be83
commit f4afc4af10
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/metadata.py
+9
View File
@@ -23,7 +23,12 @@ def get_sha256(filepath):
return hash_cache[filepath] return hash_cache[filepath]
class MetadataParser(ABC): class MetadataParser(ABC):
@abstractmethod
def get_scheme(self) -> MetadataScheme:
raise NotImplementedError
@abstractmethod @abstractmethod
def parse_json(self, metadata: dict) -> dict: def parse_json(self, metadata: dict) -> dict:
raise NotImplementedError raise NotImplementedError
@@ -34,6 +39,8 @@ class MetadataParser(ABC):
class A1111MetadataParser(MetadataParser): class A1111MetadataParser(MetadataParser):
def get_scheme(self) -> MetadataScheme:
return MetadataScheme.A1111
fooocus_to_a1111 = { fooocus_to_a1111 = {
'negative_prompt': 'Negative prompt', 'negative_prompt': 'Negative prompt',
@@ -184,6 +191,8 @@ class A1111MetadataParser(MetadataParser):
class FooocusMetadataParser(MetadataParser): class FooocusMetadataParser(MetadataParser):
def get_scheme(self) -> MetadataScheme:
return MetadataScheme.FOOOCUS
def parse_json(self, metadata: dict) -> dict: def parse_json(self, metadata: dict) -> dict:
model_filenames = modules.config.model_filenames.copy() model_filenames = modules.config.model_filenames.copy()