Spaces:

banao-tech
/

omniapi

Running on T4

[email protected] commited on 28 days ago

Commit

750f1e0

1 Parent(s): 29f706c

miss fix

Files changed (2) hide show

.DS_Store CHANGED Viewed

Binary files a/.DS_Store and b/.DS_Store differ

main.py CHANGED Viewed

@@ -42,6 +42,9 @@ import torch
 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if device == "cuda" else torch.float32  # Use float32 on CPU
 try:
     model = AutoModelForCausalLM.from_pretrained(

 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if device == "cuda" else torch.float32  # Use float32 on CPU
+processor = AutoProcessor.from_pretrained(
+    "microsoft/Florence-2-base", trust_remote_code=True
+)
 try:
     model = AutoModelForCausalLM.from_pretrained(