awinml commited on
Commit
35b8da2
β€’
1 Parent(s): 08d9a62

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ model = AutoModelForCausalLM.from_pretrained(
9
  trust_remote_code=True,
10
  device_map="auto",
11
  low_cpu_mem_usage=True,
12
- offload_folder="/",
13
  )
14
  tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-7b-instruct")
15
 
 
9
  trust_remote_code=True,
10
  device_map="auto",
11
  low_cpu_mem_usage=True,
12
+ #offload_folder="/model_files",
13
  )
14
  tokenizer = AutoTokenizer.from_pretrained("tiiuae/falcon-7b-instruct")
15