yerang commited on
Commit
f55c77d
1 Parent(s): 9d6f8ad

Update stf_utils.py

Browse files
Files changed (1) hide show
  1. 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=240)
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=240)
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(