22

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import os, random, json, spaces, torch, time, subprocess
 
 import torch
-from transformers import AutoProcessor
+from transformers import AutoProcessor, AutoTokenizer
 from longcat_image.models import LongCatImageTransformer2DModel
 from longcat_image.pipelines import LongCatImagePipeline
 
@@ -21,7 +21,7 @@ from utils.prompt_utils import polish_prompt
 MODEL_REPO = "meituan-longcat/LongCat-Image"
 MAX_SEED = np.iinfo(np.int32).max
 
-text_processor = AutoProcessor.from_pretrained( 
+text_processor = AutoTokenizer.from_pretrained( 
     MODEL_REPO, 
     subfolder = 'tokenizer'
 )