chandhron commited on
Commit
d5dbfd1
β€’
1 Parent(s): 05a9876

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ tuned_adapter = "newronai/llama-2-7b-QLoRA-Trial1"
15
  config = PeftConfig.from_pretrained(tuned_adapter)
16
  model = AutoModelForCausalLM.from_pretrained(base_model,
17
  use_cache="cache",
18
-
19
  # quantization_config=bnb_config
20
  )
21
 
 
15
  config = PeftConfig.from_pretrained(tuned_adapter)
16
  model = AutoModelForCausalLM.from_pretrained(base_model,
17
  use_cache="cache",
18
+ load_in_8bit = True
19
  # quantization_config=bnb_config
20
  )
21