# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("knoxcs/stable-diffusion-webui-knox")