yeq6x commited on
Commit
8da9921
·
1 Parent(s): fae7447
Files changed (1) hide show
  1. scripts/generate_prompt.py +1 -0
scripts/generate_prompt.py CHANGED
@@ -41,6 +41,7 @@ def download_model_files(repo_id, model_dir, sub_dir, files, sub_files):
41
  for file in sub_files:
42
  hf_hub_download(repo_id, file, subfolder=sub_dir, cache_dir=os.path.join(model_dir, sub_dir), force_download=True, force_filename=file)
43
 
 
44
  def load_wd14_tagger_model():
45
  """WD14タグ付けモデルをロード"""
46
  model_dir = "wd14_tagger_model"
 
41
  for file in sub_files:
42
  hf_hub_download(repo_id, file, subfolder=sub_dir, cache_dir=os.path.join(model_dir, sub_dir), force_download=True, force_filename=file)
43
 
44
+ @spaces.GPU
45
  def load_wd14_tagger_model():
46
  """WD14タグ付けモデルをロード"""
47
  model_dir = "wd14_tagger_model"