shekharp77 commited on
Commit
9ec47ac
1 Parent(s): 5b22c00

first commit

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -0
  2. main.py +0 -2
Dockerfile CHANGED
@@ -3,6 +3,7 @@
3
 
4
  FROM python:3.9
5
 
 
6
 
7
  WORKDIR /code
8
 
 
3
 
4
  FROM python:3.9
5
 
6
+ export TRANSFORMERS_CACHE=/code
7
 
8
  WORKDIR /code
9
 
main.py CHANGED
@@ -1,8 +1,6 @@
1
  from transformers import AutoTokenizer, LlamaForCausalLM
2
  from fastapi import FastAPI
3
  from pydantic import BaseModel
4
- import os
5
- os.environ['TRANSFORMERS_CACHE'] = '/code'
6
 
7
  model_path = 'EleutherAI/llemma_7b'
8
 
 
1
  from transformers import AutoTokenizer, LlamaForCausalLM
2
  from fastapi import FastAPI
3
  from pydantic import BaseModel
 
 
4
 
5
  model_path = 'EleutherAI/llemma_7b'
6