Browse Source

Linting error fixed

#814
7SOMAY haixuantao 10 months ago
parent
commit
67e656a72a
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      node-hub/dora-phi4/dora_phi4/main.py

+ 5
- 1
node-hub/dora-phi4/dora_phi4/main.py View File

@@ -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():


Loading…
Cancel
Save