Spaces:

kouki321
/

exemple3

Sleeping

kouki321 commited on May 21

Commit

be2b39e

verified ·

1 Parent(s): 03fbdfd

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -8,7 +8,7 @@ WORKDIR /app
 COPY requirements.txt .
 RUN pip install -r requirements.txt
-RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; AutoModelForCausalLM.from_pretrained('tiiuae/falcon-rw-1b'); AutoTokenizer.from_pretrained('tiiuae/falcon-rw-1b')"
 COPY . .

 COPY requirements.txt .
 RUN pip install -r requirements.txt
+RUN python -c "from transformers import AutoModelForCausalLM, AutoTokenizer; AutoModelForCausalLM.from_pretrained('Locutusque/TinyMistral-248M'); AutoTokenizer.from_pretrained('Locutusque/TinyMistral-248M')"
 COPY . .