gregorlied commited on
Commit
7a9d8b4
·
verified ·
1 Parent(s): 6b4f26c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -9,6 +9,9 @@ from huggingface_hub import login as hf_login
9
  from vllm import LLM
10
  from pydantic import BaseModel
11
 
 
 
 
12
  hf_login(token=os.getenv("HF_TOKEN"))
13
 
14
  model_name = "meta-llama/Llama-3.2-1B-Instruct"
 
9
  from vllm import LLM
10
  from pydantic import BaseModel
11
 
12
+ os.environ["VLLM_LOGGING_LEVEL"]="DEBUG"
13
+ os.environ["VLLM_WORKER_MULTIPROC_METHOD"]="spawn"
14
+
15
  hf_login(token=os.getenv("HF_TOKEN"))
16
 
17
  model_name = "meta-llama/Llama-3.2-1B-Instruct"