srimanth-d
/

GOT_CPU

Image-Text-to-Text

feature-extraction

vision-language

Model card Files Files and versions Community

srimanth-d commited on Dec 12, 2024

Commit

8e9b71a

·

verified ·

1 Parent(s): 4a76e0f

Update modeling_got.py

Files changed (1) hide show

modeling_got.py +1 -1

modeling_got.py CHANGED Viewed

@@ -18,7 +18,7 @@ DEFAULT_IMAGE_TOKEN = "<image>"
 DEFAULT_IMAGE_PATCH_TOKEN = '<imgpad>'
 DEFAULT_IM_START_TOKEN = '<img>'
 DEFAULT_IM_END_TOKEN = '</img>'
-device = "mps"#"cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
 print("Using device ",device)
 from enum import auto, Enum

 DEFAULT_IMAGE_PATCH_TOKEN = '<imgpad>'
 DEFAULT_IM_START_TOKEN = '<img>'
 DEFAULT_IM_END_TOKEN = '</img>'
+device = "cpu"#"cuda" if torch.cuda.is_available() else "mps" if torch.backends.mps.is_available() else "cpu"
 print("Using device ",device)
 from enum import auto, Enum