lea97338 commited on
Commit
9fa52ef
·
verified ·
1 Parent(s): 3d6dd2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,8 +24,8 @@ logger = logging.getLogger("mistral-text-encoding-gradio")
24
  # ------------------------------------------------------
25
  # Config
26
  # ------------------------------------------------------
27
- TEXT_ENCODER_ID="Qwen/Qwen2.5-1.5B-Instruct"
28
- TOKENIZER_ID = "Qwen/Qwen2.5-1.5B-Instruct"
29
  DTYPE = torch.bfloat16
30
 
31
  # ------------------------------------------------------
 
24
  # ------------------------------------------------------
25
  # Config
26
  # ------------------------------------------------------
27
+ TEXT_ENCODER_ID="Qwen/Qwen2.5-0.5B-Instruct"
28
+ TOKENIZER_ID = "Qwen/Qwen2.5-0.5B-Instruct"
29
  DTYPE = torch.bfloat16
30
 
31
  # ------------------------------------------------------