From 447a4a7fba497139bf7e52d7e77e99c8394dc0a9 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Fri, 30 Aug 2024 17:00:31 -0700
Subject: [PATCH] expose more parameters to space

---
 modules_forge/forge_space.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/modules_forge/forge_space.py b/modules_forge/forge_space.py
index 8a1d24e9..ca73795b 100644
--- a/modules_forge/forge_space.py
+++ b/modules_forge/forge_space.py
@@ -77,7 +77,7 @@ def remove_dir(dir_path):
 
 
 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.root_path = root_path
         self.hf_path = os.path.join(root_path, 'huggingface_space_mirror')
@@ -87,6 +87,9 @@ class ForgeSpace:
         self.is_running = False
         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.btn_launch = gr.Button('Launch', 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,
             local_dir=self.hf_path,
             force_download=True,
+            allow_patterns=self.allow_patterns,
+            ignore_patterns=self.ignore_patterns
         )
 
         print(f'Downloaded: {downloaded}')