mamkkl commited on
Commit
3144bc5
·
verified ·
1 Parent(s): 3c497ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -74,6 +74,7 @@ def loadModel():
74
  torch_dtype=torch.float16,
75
  cache_dir=cache_dir,
76
  device_map="auto",
 
77
  )
78
  #model = PeftModel.from_pretrained(
79
  # model,
 
74
  torch_dtype=torch.float16,
75
  cache_dir=cache_dir,
76
  device_map="auto",
77
+ load_in_8bit=True
78
  )
79
  #model = PeftModel.from_pretrained(
80
  # model,