# class MlApiConfig(AppConfig): # default_auto_field = "django.db.models.BigAutoField" # name = "ml_api" # tokenizer = None # model = None # def ready(self): # if MlApiConfig.model is None: # model_name = "rinna/japanese-gpt-neox-small" # print("Loading Model... (Download starts on first run)") # # トークナイザーとモデルのロード # MlApiConfig.tokenizer = AutoTokenizer.from_pretrained( # model_name, use_fast=False # ) # MlApiConfig.model = AutoModelForCausalLM.from_pretrained(model_name) # print("Model Loaded!") from django.apps import AppConfig class MlApiConfig(AppConfig): default_auto_field = "django.db.models.BigAutoField" name = "ml_api"