Update preload.py

justinwlin · Feb 18, 2024 · 7daa19a · 7daa19a
1 parent 1b81c35
commit 7daa19a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/preload.py b/preload.py
@@ -4,4 +4,6 @@
 # Use the MODEL environment variable; fallback to a default if not set
 model_name = os.getenv('MODEL', 'mistralai/Mistral-7B-Instruct-v0.1')
 
+print("Model name: ", model_name)
+
 llm = openllm.LLM(model_name, backend="vllm")