mamkkl commited on
Commit
549ebd5
·
verified ·
1 Parent(s): d811f10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -99,7 +99,7 @@ def loadModel():
99
  torch_dtype=torch.float16,
100
  cache_dir=cache_dir,
101
  device_map="auto",
102
- quantization_config=quantization_config,
103
  max_memory={
104
  0: "20GB", # GPU 0 with 20GB memory
105
  1: "40GB", # GPU 0 with 20GB memory
 
99
  torch_dtype=torch.float16,
100
  cache_dir=cache_dir,
101
  device_map="auto",
102
+ #quantization_config=quantization_config,
103
  max_memory={
104
  0: "20GB", # GPU 0 with 20GB memory
105
  1: "40GB", # GPU 0 with 20GB memory