check.py 1.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. def check_model_info(self, version_uuid):
  2. """查询模型详细信息,确认其基础算法(baseAlgo)"""
  3. uri = "/api/model/version/get"
  4. url = self._generate_signature(uri)
  5. payload = {
  6. "versionUuid": version_uuid
  7. }
  8. try:
  9. print(f"🔍 正在查询模型信息: {version_uuid}...")
  10. resp = requests.post(
  11. url,
  12. headers={"Content-Type": "application/json"},
  13. json=payload,
  14. timeout=10
  15. )
  16. result = resp.json()
  17. if result.get("code") == 0:
  18. data = result.get("data", {})
  19. print(f"--- 模型信息 ---")
  20. print(f"📌 模型名称: {data.get('model_name')}")
  21. print(f"📌 版本名称: {data.get('version_name')}")
  22. print(f"🚀 基础算法 (baseAlgo): {data.get('baseAlgo')}") # 重点看这个!
  23. print(f"💰 是否商用: {'是' if data.get('commercial_use') == '1' else '否'}")
  24. print(f"----------------")
  25. return data
  26. else:
  27. print(f"❌ 查询失败: {result.get('msg')}")
  28. return None
  29. except Exception as e:
  30. print(f"❌ 请求异常: {str(e)}")
  31. return None
  32. # --- 在 main 中使用 ---
  33. if __name__ == "__main__":
  34. client = LibLibControlNet()
  35. # 1. 先查底模
  36. print("检查底模架构...")
  37. client.check_model_info(client.CHECKPOINT_ID)
  38. # 2. 再查 ControlNet 模型
  39. print("\n检查 ControlNet 模型架构...")
  40. client.check_model_info(client.CANNY_MODEL_ID)