AnishHF commited on
Commit
ac1a107
1 Parent(s): 0fe2940

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ access_token = os.environ["GATED_ACCESS_TOKEN"]
9
  quantization_config = BitsAndBytesConfig(
10
  load_in_4bit=True,
11
  bnb_4bit_quant_type="nf4",
12
- bnb_4bit_compute_dtype="torch.int8,
13
  )
14
 
15
  model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", quantization_config=True, device_map="auto", token=access_token)
 
9
  quantization_config = BitsAndBytesConfig(
10
  load_in_4bit=True,
11
  bnb_4bit_quant_type="nf4",
12
+ bnb_4bit_compute_dtype="torch.int8",
13
  )
14
 
15
  model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1", quantization_config=True, device_map="auto", token=access_token)