John Doe
commited on
Commit
·
18f66fe
1
Parent(s):
dd74f4c
quick
Browse files- modules/app.py +1 -1
modules/app.py
CHANGED
|
@@ -56,7 +56,7 @@ app = FastAPI(docs_url=None, redoc_url=None)
|
|
| 56 |
app.mount("/static", StaticFiles(directory="static"), name="static")
|
| 57 |
|
| 58 |
|
| 59 |
-
logging.info("loading model")
|
| 60 |
model = AutoModelForCausalLM.from_pretrained(model_name, **kwargs)
|
| 61 |
logging.info("loading tokenizer")
|
| 62 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
|
|
|
| 56 |
app.mount("/static", StaticFiles(directory="static"), name="static")
|
| 57 |
|
| 58 |
|
| 59 |
+
logging.info(f"loading model {model_name}")
|
| 60 |
model = AutoModelForCausalLM.from_pretrained(model_name, **kwargs)
|
| 61 |
logging.info("loading tokenizer")
|
| 62 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|