Use Maia 23m as default model
This commit is contained in:
@@ -8,7 +8,7 @@ Type=simple
|
|||||||
User=root
|
User=root
|
||||||
Group=root
|
Group=root
|
||||||
WorkingDirectory=/root/blog-backend/maia-engine
|
WorkingDirectory=/root/blog-backend/maia-engine
|
||||||
Environment=MAIA_MODEL=maia3-5m
|
Environment=MAIA_MODEL=maia3-23m
|
||||||
Environment=MAIA_DEVICE=cpu
|
Environment=MAIA_DEVICE=cpu
|
||||||
Environment=MAIA_USE_AMP=false
|
Environment=MAIA_USE_AMP=false
|
||||||
Environment=HF_HOME=/root/blog-backend/.cache/huggingface
|
Environment=HF_HOME=/root/blog-backend/.cache/huggingface
|
||||||
|
|||||||
@@ -49,7 +49,7 @@ services:
|
|||||||
container_name: maia-engine
|
container_name: maia-engine
|
||||||
restart: always
|
restart: always
|
||||||
environment:
|
environment:
|
||||||
- MAIA_MODEL=maia3-5m
|
- MAIA_MODEL=maia3-23m
|
||||||
- MAIA_DEVICE=cpu
|
- MAIA_DEVICE=cpu
|
||||||
- MAIA_USE_AMP=false
|
- MAIA_USE_AMP=false
|
||||||
- HF_HOME=/models/huggingface
|
- HF_HOME=/models/huggingface
|
||||||
|
|||||||
@@ -13,7 +13,7 @@ from fastapi import FastAPI, HTTPException
|
|||||||
from pydantic import BaseModel, Field
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
DEFAULT_MODEL = os.getenv("MAIA_MODEL", "5m").removeprefix("maia3-")
|
DEFAULT_MODEL = os.getenv("MAIA_MODEL", "23m").removeprefix("maia3-")
|
||||||
DEFAULT_DEVICE = os.getenv("MAIA_DEVICE", "cpu")
|
DEFAULT_DEVICE = os.getenv("MAIA_DEVICE", "cpu")
|
||||||
DEFAULT_USE_AMP = os.getenv("MAIA_USE_AMP", "false").lower() == "true"
|
DEFAULT_USE_AMP = os.getenv("MAIA_USE_AMP", "false").lower() == "true"
|
||||||
|
|
||||||
|
|||||||
@@ -25,7 +25,7 @@ data class ChessGameCreateRequest(
|
|||||||
regexp = "^(3m|5m|23m|79m)$",
|
regexp = "^(3m|5m|23m|79m)$",
|
||||||
message = "model은 3m, 5m, 23m, 79m 중 하나여야 합니다."
|
message = "model은 3m, 5m, 23m, 79m 중 하나여야 합니다."
|
||||||
)
|
)
|
||||||
val model: String = "5m",
|
val model: String = "23m",
|
||||||
|
|
||||||
@field:DecimalMin(value = "0.0", message = "temperature는 0.0 이상이어야 합니다.")
|
@field:DecimalMin(value = "0.0", message = "temperature는 0.0 이상이어야 합니다.")
|
||||||
@field:DecimalMax(value = "2.0", message = "temperature는 2.0 이하여야 합니다.")
|
@field:DecimalMax(value = "2.0", message = "temperature는 2.0 이하여야 합니다.")
|
||||||
|
|||||||
Reference in New Issue
Block a user