farrell236 commited on
Commit
256a744
·
1 Parent(s): 48851c3
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ import torch
14
  from qwen_vl_utils import process_vision_info
15
  from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration, TextIteratorStreamer
16
 
17
- DEFAULT_CKPT_PATH = 'farrell236/test_model'
18
  # DEFAULT_CKPT_PATH = '/scratch/llm-weights/Qwen/Qwen2.5-VL-7B-Instruct'
19
  AUTH_TOKEN = os.environ.get("HF_spaces")
20
 
@@ -71,7 +71,7 @@ def _load_model_processor(args):
71
  torch_dtype=torch.bfloat16,
72
  device_map=device_map)
73
 
74
- processor = AutoProcessor.from_pretrained('Qwen/Qwen2.5-VL-3B-Instruct')
75
  return model, processor
76
 
77
 
 
14
  from qwen_vl_utils import process_vision_info
15
  from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration, TextIteratorStreamer
16
 
17
+ DEFAULT_CKPT_PATH = 'farrell236/OpthModel32B_b'
18
  # DEFAULT_CKPT_PATH = '/scratch/llm-weights/Qwen/Qwen2.5-VL-7B-Instruct'
19
  AUTH_TOKEN = os.environ.get("HF_spaces")
20
 
 
71
  torch_dtype=torch.bfloat16,
72
  device_map=device_map)
73
 
74
+ processor = AutoProcessor.from_pretrained('Qwen/Qwen2.5-VL-32B-Instruct')
75
  return model, processor
76
 
77