Mikhil-jivus commited on
Commit
3f23cc2
·
verified ·
1 Parent(s): 5faed4e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ model = AutoModelForCausalLM.from_pretrained(
16
  repo_id,
17
  token=access_token,
18
  torch_dtype=torch.bfloat16, # or use torch.bfloat16 if supported
19
- device_map="auto" # Automatically use available GPU/CPU efficiently
 
20
  )
21
 
22
  def respond(
 
16
  repo_id,
17
  token=access_token,
18
  torch_dtype=torch.bfloat16, # or use torch.bfloat16 if supported
19
+ device_map="auto", # Automatically use available GPU/CPU efficiently
20
+ low_cpu_mem_usage=True
21
  )
22
 
23
  def respond(