MuhammadNoman7600 commited on
Commit
f478d14
Β·
verified Β·
1 Parent(s): 434ecaa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
25
  MODEL_NAME = "Qwen/Qwen2.5-0.5B-Instruct"
26
  HOST = "0.0.0.0"
27
  PORT = 7860
28
- MAX_NEW_TOKENS = 1024
29
  # ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
30
 
31
  app = FastAPI(
 
25
  MODEL_NAME = "Qwen/Qwen2.5-0.5B-Instruct"
26
  HOST = "0.0.0.0"
27
  PORT = 7860
28
+ MAX_NEW_TOKENS = 32768
29
  # ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━
30
 
31
  app = FastAPI(