diff --git a/node-hub/dora-phi4/dora_phi4/main.py b/node-hub/dora-phi4/dora_phi4/main.py index 0944a08a..6cfc5067 100644 --- a/node-hub/dora-phi4/dora_phi4/main.py +++ b/node-hub/dora-phi4/dora_phi4/main.py @@ -8,7 +8,11 @@ import torch from accelerate import infer_auto_device_map from dora import Node from PIL import Image -from transformers import AutoModelForCausalLM, AutoProcessor, GenerationConfig +from transformers import ( + AutoModelForCausalLM, + AutoProcessor, + GenerationConfig, +) # 🔍 Detect the best available device if torch.cuda.is_available():