Update app.py
Browse files
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
|