Spaces:

xiezhe22
/

ChatTS

Running on Zero

xiezhe22 commited on Jul 24

Commit

009e745

1 Parent(s): 77babc6

FIx

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ model.eval()
 # ─── INFERENCE + VALIDATION ────────────────────────────────────────────────────
-@spaces.gpu
 def generate_text(prompt):
     inputs = tokenizer([prompt], return_tensors="pt").to(model.device)
     outputs = model.generate(

 # ─── INFERENCE + VALIDATION ────────────────────────────────────────────────────
+@spaces.GPU
 def generate_text(prompt):
     inputs = tokenizer([prompt], return_tensors="pt").to(model.device)
     outputs = model.generate(