Linseypass commited on
Commit
367f1e8
·
1 Parent(s): 155a93f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ model_name = "decapoda-research/llama-7b-hf"
15
  adapters_name = 'timdettmers/guanaco-7b'
16
  # print(f"Starting to load the model {model_name} into memory")
17
  m = AutoModelForCausalLM.from_pretrained(
18
- model_name)
 
19
  m = PeftModel.from_pretrained(m, adapters_name)
20
  m = m.merge_and_unload()
21
  tok = LlamaTokenizer.from_pretrained(model_name)
 
15
  adapters_name = 'timdettmers/guanaco-7b'
16
  # print(f"Starting to load the model {model_name} into memory")
17
  m = AutoModelForCausalLM.from_pretrained(
18
+ model_name,
19
+ load_in_4bit=True)
20
  m = PeftModel.from_pretrained(m, adapters_name)
21
  m = m.merge_and_unload()
22
  tok = LlamaTokenizer.from_pretrained(model_name)