Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -118,8 +118,8 @@ try:
|
|
118 |
base_model_path,
|
119 |
torch_dtype=torch.float16,
|
120 |
device_map="cpu", # Explicitly set CPU
|
121 |
-
load_in_8bit=True # Enable int8 quantization
|
122 |
-
|
123 |
# device_map="auto"
|
124 |
)
|
125 |
|
|
|
118 |
base_model_path,
|
119 |
torch_dtype=torch.float16,
|
120 |
device_map="cpu", # Explicitly set CPU
|
121 |
+
# load_in_8bit=True # Enable int8 quantization
|
122 |
+
trust_remote_code=True,
|
123 |
# device_map="auto"
|
124 |
)
|
125 |
|