lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-27 01:38:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Forge Space and BiRefNet

Browse Source
This commit is contained in:
lllyasviel
2024-08-17 08:29:08 -07:00
committed by GitHub
parent 8a04293430
commit 93b40f355e
13 changed files with 390 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules/extensions.py
View File

@@ -5,6 +5,7 @@ import dataclasses
import os
import threading
import re
import json
from modules import shared, errors, cache, scripts
from modules.gitpython_hack import Repo
@@ -124,6 +125,13 @@ class Extension:
self.metadata = metadata if metadata else ExtensionMetadata(self.path, name.lower())
self.canonical_name = metadata.canonical_name
self.is_forge_space = False
self.space_meta = None
if os.path.exists(os.path.join(self.path, 'space_meta.json')) and os.path.exists(os.path.join(self.path, 'forge_app.py')):
self.is_forge_space = True
self.space_meta = json.load(open(os.path.join(self.path, 'space_meta.json'), 'rt', encoding='utf-8'))
def to_dict(self):
return {x: getattr(self, x) for x in self.cached_fields}