fix errors

This commit is contained in:
Xingchao Liu 2024-11-13 11:49:24 +08:00
parent f887e7270e
commit 146668eafe

View File

@ -9,7 +9,6 @@ cuda_device = 'cuda' if torch.cuda.is_available() else 'cpu'
# Load model and processor # Load model and processor
model_path = "deepseek-ai/JanusFlow-1.3B" model_path = "deepseek-ai/JanusFlow-1.3B"
model_path = "/weka-jd/prod/jupyter/liuxingchao/notebooks/janus/final_converted_ckpt_ema"
vl_chat_processor = VLChatProcessor.from_pretrained(model_path) vl_chat_processor = VLChatProcessor.from_pretrained(model_path)
tokenizer = vl_chat_processor.tokenizer tokenizer = vl_chat_processor.tokenizer