Update app.py
Browse files
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"
|