credent007 commited on
Commit
05759c9
·
verified ·
1 Parent(s): 502e4a5

Update model_loader.py

Browse files
Files changed (1) hide show
  1. model_loader.py +0 -33
model_loader.py CHANGED
@@ -22,36 +22,3 @@ model = AutoModelForVision2Seq.from_pretrained(
22
  print("Model loaded successfully")
23
 
24
 
25
-
26
- # import torch
27
- # from transformers import AutoProcessor, AutoModelForVision2Seq
28
-
29
- # MODEL_NAME = "Qwen/Qwen2.5-VL-7B-Instruct"
30
-
31
- # model = None
32
- # processor = None
33
- # device = "cuda" if torch.cuda.is_available() else "cpu"
34
-
35
-
36
- # def get_model():
37
- # global model, processor, device
38
-
39
- # if model is None or processor is None:
40
- # print("Loading processor...")
41
- # processor = AutoProcessor.from_pretrained(
42
- # MODEL_NAME,
43
- # trust_remote_code=True,
44
- # use_fast=True
45
- # )
46
-
47
- # print("Loading model...")
48
- # model = AutoModelForVision2Seq.from_pretrained(
49
- # MODEL_NAME,
50
- # trust_remote_code=True,
51
- # torch_dtype=torch.float16,
52
- # device_map="auto"
53
- # )
54
-
55
- # print("Model loaded successfully")
56
-
57
- # return model, processor, device
 
22
  print("Model loaded successfully")
23
 
24