Spaces:

eltorio
/

IDEFICS3_ROCO

Running

æLtorio commited on 9 days ago

Commit

ca9f0b9

•

1 Parent(s): db6d984

fallback to cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import AutoProcessor, Idefics3ForConditionalGeneration, image_utils
 import torch
-device = torch.device('cuda') # if torch.cuda.is_available() else torch.device('cpu')
 print(f"Using device: {device}")
 model_id="eltorio/IDEFICS3_ROCO"
 # model = AutoModelForImageTextToText.from_pretrained(model_id).to(device)

 import gradio as gr
 from transformers import AutoProcessor, Idefics3ForConditionalGeneration, image_utils
 import torch
+device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 print(f"Using device: {device}")
 model_id="eltorio/IDEFICS3_ROCO"
 # model = AutoModelForImageTextToText.from_pretrained(model_id).to(device)