rphrp1985 commited on
Commit
31e4426
·
verified ·
1 Parent(s): ccb1159

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ accelerator = Accelerator()
61
  model = AutoModelForCausalLM.from_pretrained(model_id, token= token,
62
  # torch_dtype= torch.uint8,
63
  torch_dtype=torch.bfloat16,
64
- load_in_8bit=True,
65
  # # # torch_dtype=torch.fl,
66
  attn_implementation="flash_attention_2",
67
  low_cpu_mem_usage=True,
 
61
  model = AutoModelForCausalLM.from_pretrained(model_id, token= token,
62
  # torch_dtype= torch.uint8,
63
  torch_dtype=torch.bfloat16,
64
+ load_in_4bit=True,
65
  # # # torch_dtype=torch.fl,
66
  attn_implementation="flash_attention_2",
67
  low_cpu_mem_usage=True,