Frederic-CellNum commited on
Commit
094dfca
·
verified ·
1 Parent(s): 7f6ae1b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -3
Dockerfile CHANGED
@@ -7,9 +7,10 @@ COPY requirements.txt .
7
  RUN pip install -r requirements.txt
8
 
9
  # Pré-télécharger le modèle PENDANT le build (accès réseau autorisé)
10
- RUN python -c "from transformers import Qwen2VLForConditionalGeneration, AutoProcessor\
11
- AutoProcessor.from_pretrained('Qwen/Qwen2-VL-7B-Instruct')\
12
- Qwen2VLForConditionalGeneration.from_pretrained('Qwen/Qwen2-VL-7B-Instruct', torch_dtype='auto')
 
13
  "
14
 
15
  COPY . .
 
7
  RUN pip install -r requirements.txt
8
 
9
  # Pré-télécharger le modèle PENDANT le build (accès réseau autorisé)
10
+ RUN python -c "\
11
+ from transformers import Qwen2VLForConditionalGeneration, AutoProcessor; \
12
+ AutoProcessor.from_pretrained('Qwen/Qwen2-VL-7B-Instruct'); \
13
+ Qwen2VLForConditionalGeneration.from_pretrained('Qwen/Qwen2-VL-7B-Instruct', torch_dtype='auto') \
14
  "
15
 
16
  COPY . .