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)}")
|