【DEEPSEEK】Janus-Pro-1B模型推理、官方demo代码以及流式输出

官方代码地址：GitCode - 全球开发者的开源社区,开源代码托管平台

目录

1 fastapi_app.py

2 fastapi_client.py

3 推理

---

---

1 fastapi_app.py

from fastapi import FastAPI, File, Form, UploadFile, HTTPException
from fastapi.responses import JSONResponse, StreamingResponse
import torch
from transformers import AutoConfig, AutoModelForCausalLM
from janus.models import MultiModalityCausalLM, VLChatProcessor
from PIL import Image
import numpy as np
import io

app = FastAPI()

# Load model