Spaces:
Running
on
Zero
Running
on
Zero
Update stf_utils.py
Browse files- stf_utils.py +2 -2
stf_utils.py
CHANGED
@@ -91,7 +91,7 @@ class STFPipeline:
|
|
91 |
self.model = self.load_model()
|
92 |
self.template = self.create_template()
|
93 |
|
94 |
-
@spaces.GPU(duration=
|
95 |
def load_model(self):
|
96 |
"""모델을 생성하고 GPU에 할당."""
|
97 |
model = stf_alternative.create_model(
|
@@ -103,7 +103,7 @@ class STFPipeline:
|
|
103 |
)
|
104 |
return model
|
105 |
|
106 |
-
@spaces.GPU(duration=
|
107 |
def create_template(self):
|
108 |
"""템플릿 생성."""
|
109 |
template = stf_alternative.Template(
|
|
|
91 |
self.model = self.load_model()
|
92 |
self.template = self.create_template()
|
93 |
|
94 |
+
@spaces.GPU(duration=120)
|
95 |
def load_model(self):
|
96 |
"""모델을 생성하고 GPU에 할당."""
|
97 |
model = stf_alternative.create_model(
|
|
|
103 |
)
|
104 |
return model
|
105 |
|
106 |
+
@spaces.GPU(duration=120)
|
107 |
def create_template(self):
|
108 |
"""템플릿 생성."""
|
109 |
template = stf_alternative.Template(
|