Hugging Face's logo

Spaces:
adaptiveaiventures
/
llama2-interference
Runtime error

App Files Community
adaptiveaiventures commited on
Commit
7559f23
·
verified ·
1 Parent(s): b3fc828

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -11,4 +11,4 @@ ENV TRANSFORMERS_CACHE="/tmp"
11
  ENV HUGGINGFACE_HUB_CACHE="/tmp"
12
 
13
  # Run the TGI server
14
- CMD ["--model-id", "adaptiveaiventures/Llama-2-7b-chat-finetune", "--port", "8080", "--num-shard", "1", "--dtype", "bfloat16", "--max-batch-prefill-tokens", "512", "--disable-custom-kernels"]
 
11
  ENV HUGGINGFACE_HUB_CACHE="/tmp"
12
 
13
  # Run the TGI server
14
+ CMD ["--model-id", "adaptiveaiventures/Llama-32-1B-finetuned-aaivtest5", "--port", "8080", "--num-shard", "1", "--dtype", "bfloat16", "--max-batch-prefill-tokens", "512", "--disable-custom-kernels"]