Spaces:
Sleeping
Sleeping
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 |
|