deepseekmirror/Janus
1
0
Fork 0
mirror of https://github.com/deepseek-ai/Janus.git synced 2025-02-23 06:08:59 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Adds MODEL_NAME as an Environment variable for Docker context. Defaults to the same 1.3B model

Browse Source
This commit is contained in:
julian 2025-01-28 12:49:39 -05:00
parent e471403e44
commit bfa6c0e1c1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
demo/fastapi_app.py
View File

@ -6,11 +6,12 @@ from janus.models import MultiModalityCausalLM, VLChatProcessor
from PIL import Image from PIL import Image
import numpy as np import numpy as np
import io import io
import os
app = FastAPI() app = FastAPI()
# Load model and processor # Load model and processor
model_path = "deepseek-ai/Janus-1.3B" model_path = os.getenv("MODEL_NAME", "deepseek-ai/Janus-1.3B")
config = AutoConfig.from_pretrained(model_path) config = AutoConfig.from_pretrained(model_path)
language_config = config.language_config language_config = config.language_config
language_config._attn_implementation = 'eager' language_config._attn_implementation = 'eager'