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

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -5
Dockerfile CHANGED
@@ -7,11 +7,9 @@ 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 "\
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 . .
17
 
 
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
+ RUN python -c "AutoProcessor.from_pretrained('Qwen/Qwen2-VL-7B-Instruct')"
12
+ RUN python -c "Qwen2VLForConditionalGeneration.from_pretrained('Qwen/Qwen2-VL-7B-Instruct', torch_dtype='auto')"
 
 
13
 
14
  COPY . .
15