mistral-text-encoder

Running on Zero

Gemini899 commited on 30 days ago

Commit

c0eaed0

verified ·

1 Parent(s): bb472aa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ def get_vram_info():
     return {"vram": "CUDA not available"}
-@spaces.GPU(duration=15)
 def encode_text(prompt: str):
     """Encode text and return a downloadable pytorch file."""
     global text_encoder, tokenizer

     return {"vram": "CUDA not available"}
+@spaces.GPU(duration=18)
 def encode_text(prompt: str):
     """Encode text and return a downloadable pytorch file."""
     global text_encoder, tokenizer