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

expose more parameters to space

Browse Source
This commit is contained in:
layerdiffusion
2024-08-30 17:00:31 -07:00
parent 8e9b124456
commit 447a4a7fba
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules_forge/forge_space.py
View File

@@ -77,7 +77,7 @@ def remove_dir(dir_path):
class ForgeSpace: class ForgeSpace:
def __init__(self, root_path, title, repo_id=None, repo_type='space', revision=None, **kwargs): def __init__(self, root_path, title, repo_id=None, repo_type='space', revision=None, allow_patterns=None, ignore_patterns=None, **kwargs):
self.title = title self.title = title
self.root_path = root_path self.root_path = root_path
self.hf_path = os.path.join(root_path, 'huggingface_space_mirror') self.hf_path = os.path.join(root_path, 'huggingface_space_mirror')
@@ -87,6 +87,9 @@ class ForgeSpace:
self.is_running = False self.is_running = False
self.gradio_metas = None self.gradio_metas = None
self.allow_patterns = allow_patterns
self.ignore_patterns = ignore_patterns
self.label = gr.HTML(build_html(title=title, url=None), elem_classes=['forge_space_label']) self.label = gr.HTML(build_html(title=title, url=None), elem_classes=['forge_space_label'])
self.btn_launch = gr.Button('Launch', elem_classes=['forge_space_btn']) self.btn_launch = gr.Button('Launch', elem_classes=['forge_space_btn'])
self.btn_terminate = gr.Button('Terminate', elem_classes=['forge_space_btn']) self.btn_terminate = gr.Button('Terminate', elem_classes=['forge_space_btn'])
@@ -137,6 +140,8 @@ class ForgeSpace:
revision=self.revision, revision=self.revision,
local_dir=self.hf_path, local_dir=self.hf_path,
force_download=True, force_download=True,
allow_patterns=self.allow_patterns,
ignore_patterns=self.ignore_patterns
) )
print(f'Downloaded: {downloaded}') print(f'Downloaded: {downloaded}')