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