Skip to content

Commit 2fe2764

Browse files
committed
rebase & lint
1 parent c6aa251 commit 2fe2764

File tree

1 file changed

+3
-2
lines changed
  • examples/qualcomm/oss_scripts

1 file changed

+3
-2
lines changed

examples/qualcomm/oss_scripts/deit.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010
from multiprocessing.connection import Client
1111

1212
import numpy as np
13-
import torch
1413
from executorch.backends.qualcomm._passes.qnn_pass_manager import (
1514
get_capture_program_passes,
1615
)
@@ -29,7 +28,9 @@
2928

3029
def get_instance():
3130
module = (
32-
AutoModelForImageClassification.from_pretrained("facebook/deit-base-distilled-patch16-224")
31+
AutoModelForImageClassification.from_pretrained(
32+
"facebook/deit-base-distilled-patch16-224"
33+
)
3334
.eval()
3435
.to("cpu")
3536
)

0 commit comments

Comments
 (0)