Update app.py
Browse files
app.py
CHANGED
|
@@ -24,13 +24,13 @@ NUMBA_CACHE_DIR = "/app/cache"
|
|
| 24 |
# Initialize BitNet model for CPU-only
|
| 25 |
try:
|
| 26 |
llm = LLM(
|
| 27 |
-
model="1bitLLM/bitnet_b1_58-3B",
|
| 28 |
device="cpu",
|
| 29 |
enforce_eager=True,
|
| 30 |
tensor_parallel_size=1,
|
| 31 |
disable_custom_all_reduce=True,
|
| 32 |
max_model_len=2048,
|
| 33 |
-
dtype="float32",
|
| 34 |
)
|
| 35 |
except Exception as e:
|
| 36 |
logger.error(f"Failed to load BitNet model: {str(e)}")
|
|
|
|
| 24 |
# Initialize BitNet model for CPU-only
|
| 25 |
try:
|
| 26 |
llm = LLM(
|
| 27 |
+
model="1bitLLM/bitnet_b1_58-3B",
|
| 28 |
device="cpu",
|
| 29 |
enforce_eager=True,
|
| 30 |
tensor_parallel_size=1,
|
| 31 |
disable_custom_all_reduce=True,
|
| 32 |
max_model_len=2048,
|
| 33 |
+
dtype="float32",
|
| 34 |
)
|
| 35 |
except Exception as e:
|
| 36 |
logger.error(f"Failed to load BitNet model: {str(e)}")
|