ChintanSatva commited on
Commit
9acdfd7
·
verified ·
1 Parent(s): ffd1cb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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", # Stable BitNet model; fine-tune later
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", # Ensure CPU-compatible dtype
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)}")