AmeerH commited on
Commit
c92a6d2
·
1 Parent(s): d27ca99

Update app.py to remove local files only.

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -42,12 +42,10 @@ model_path = 'meta-llama/Llama-2-7b-chat-hf'
42
 
43
  tokenizer = AutoTokenizer.from_pretrained(
44
  model_path,
45
- local_files_only=True,
46
  )
47
 
48
  model = AutoModelForCausalLM.from_pretrained(
49
  model_path,
50
- local_files_only=True,
51
  low_cpu_mem_usage=True,
52
  device_map="auto",
53
  offload_folder="offload/",
 
42
 
43
  tokenizer = AutoTokenizer.from_pretrained(
44
  model_path,
 
45
  )
46
 
47
  model = AutoModelForCausalLM.from_pretrained(
48
  model_path,
 
49
  low_cpu_mem_usage=True,
50
  device_map="auto",
51
  offload_folder="offload/",