File size: 242 Bytes
c655cdc
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
# vtoonify_model.py
import torch
from some_vtoonify_lib import VToonifyModel  # replace with actual

def load_vtoonify():
    model = VToonifyModel.from_pretrained("PKUWilliamYang/VToonify")
    model.eval()
    return lambda img: model(img)