Hugging Face's logo

Spaces:
KBlueLeaf
/
DTG-demo
Running on Zero

App Files Community
4
Kohaku-Blueleaf commited on
Commit
edff9b1
·
1 Parent(s): 41acb2e

zeroGPU try2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,6 @@ from kgen.metainfo import SPECIAL, TARGET
15
  MODEL_PATH = "KBlueLeaf/DanTagGen"
16
 
17
 
18
- @spaces.GPU
19
  @torch.no_grad()
20
  def get_result(
21
  text_model: LlamaForCausalLM,
@@ -117,6 +116,7 @@ if __name__ == "__main__":
117
  text_model = LlamaForCausalLM.from_pretrained(MODEL_PATH)
118
  text_model = text_model.eval()
119
 
 
120
  def wrapper(
121
  rating: str,
122
  artist: str,
 
15
  MODEL_PATH = "KBlueLeaf/DanTagGen"
16
 
17
 
 
18
  @torch.no_grad()
19
  def get_result(
20
  text_model: LlamaForCausalLM,
 
116
  text_model = LlamaForCausalLM.from_pretrained(MODEL_PATH)
117
  text_model = text_model.eval()
118
 
119
+ @spaces.GPU
120
  def wrapper(
121
  rating: str,
122
  artist: str,