prithivMLmods commited on
Commit
596439f
·
verified ·
1 Parent(s): 651f794

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -17,7 +17,6 @@ from PIL import Image
17
  from transformers import (
18
  Qwen2VLForConditionalGeneration,
19
  Qwen2_5_VLForConditionalGeneration,
20
- AutoModelForVision2Seq,
21
  AutoModelForImageTextToText,
22
  AutoProcessor,
23
  TextIteratorStreamer,
@@ -98,7 +97,7 @@ processor_j = AutoProcessor.from_pretrained(
98
  MODEL_ID_J,
99
  trust_remote_code=True
100
  )
101
- model_j = AutoModelForVision2Seq.from_pretrained(
102
  MODEL_ID_J,
103
  trust_remote_code=True,
104
  torch_dtype=torch.float16
 
17
  from transformers import (
18
  Qwen2VLForConditionalGeneration,
19
  Qwen2_5_VLForConditionalGeneration,
 
20
  AutoModelForImageTextToText,
21
  AutoProcessor,
22
  TextIteratorStreamer,
 
97
  MODEL_ID_J,
98
  trust_remote_code=True
99
  )
100
+ model_j = Qwen2_5_VLForConditionalGeneration.from_pretrainedd(
101
  MODEL_ID_J,
102
  trust_remote_code=True,
103
  torch_dtype=torch.float16