lea97338 commited on
Commit
3d6dd2b
·
verified ·
1 Parent(s): 18b267d

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-7B-Instruct-1M"
28
- TOKENIZER_ID = "Qwen/Qwen2.5-7B-Instruct-1M"
29
  DTYPE = torch.bfloat16
30
 
31
  # ------------------------------------------------------
 
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
  # ------------------------------------------------------