Joetib commited on
Commit
face0b6
1 Parent(s): 6f5214e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,11 +5,11 @@ app = FastAPI()
5
 
6
  model_id = "ibleducation/ibl-fordham-7b"
7
 
8
- tokenizer = AutoTokenizer.from_pretrained(model_id, cache_dir="./cache",)
9
  model = AutoModelForCausalLM.from_pretrained(
10
  model_id,
11
  device_map="auto",
12
- cache_dir="./cache",
13
  )
14
  pipeline = transformers.pipeline(
15
  "text-generation",
 
5
 
6
  model_id = "ibleducation/ibl-fordham-7b"
7
 
8
+ tokenizer = AutoTokenizer.from_pretrained(model_id, cache_dir=".",)
9
  model = AutoModelForCausalLM.from_pretrained(
10
  model_id,
11
  device_map="auto",
12
+ cache_dir=".",
13
  )
14
  pipeline = transformers.pipeline(
15
  "text-generation",