fiewolf1000 commited on
Commit
2b9f49c
·
verified ·
1 Parent(s): 3f6d97b

Update model_space.py

Browse files
Files changed (1) hide show
  1. model_space.py +7 -6
model_space.py CHANGED
@@ -13,12 +13,7 @@ import psutil
13
  # 模型加载后,添加CPU指令集日志
14
  import subprocess
15
 
16
- # 执行命令查看CPU支持的指令集
17
- try:
18
- avx2_support = subprocess.check_output("grep -c avx2 /proc/cpuinfo", shell=True).decode().strip()
19
- logger.info(f"CPU AVX2支持:{'是' if int(avx2_support) > 0 else '否'}")
20
- except Exception as e:
21
- logger.warning(f"检测AVX2支持失败:{str(e)}")
22
  # 日志与FastAPI初始化
23
  logging.basicConfig(
24
  level=logging.INFO,
@@ -30,6 +25,12 @@ logger = logging.getLogger(__name__)
30
  app = FastAPI(title="CodeLlama-7B-Instruct (GGUF-4bit) CPU", version="1.0")
31
  logger.info("FastAPI应用初始化完成")
32
 
 
 
 
 
 
 
33
  # 跨域配置
34
  app.add_middleware(
35
  CORSMiddleware,
 
13
  # 模型加载后,添加CPU指令集日志
14
  import subprocess
15
 
16
+
 
 
 
 
 
17
  # 日志与FastAPI初始化
18
  logging.basicConfig(
19
  level=logging.INFO,
 
25
  app = FastAPI(title="CodeLlama-7B-Instruct (GGUF-4bit) CPU", version="1.0")
26
  logger.info("FastAPI应用初始化完成")
27
 
28
+ # 执行命令查看CPU支持的指令集
29
+ try:
30
+ avx2_support = subprocess.check_output("grep -c avx2 /proc/cpuinfo", shell=True).decode().strip()
31
+ logger.info(f"CPU AVX2支持:{'是' if int(avx2_support) > 0 else '否'}")
32
+ except Exception as e:
33
+ logger.warning(f"检测AVX2支持失败:{str(e)}")
34
  # 跨域配置
35
  app.add_middleware(
36
  CORSMiddleware,