From 67e656a72a9a31216e2b1a95fac70763e44e65e1 Mon Sep 17 00:00:00 2001 From: 7SOMAY Date: Tue, 11 Mar 2025 17:49:04 +0530 Subject: [PATCH] Linting error fixed --- node-hub/dora-phi4/dora_phi4/main.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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():