We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c6aa251 commit 2fe2764Copy full SHA for 2fe2764
examples/qualcomm/oss_scripts/deit.py
@@ -10,7 +10,6 @@
10
from multiprocessing.connection import Client
11
12
import numpy as np
13
-import torch
14
from executorch.backends.qualcomm._passes.qnn_pass_manager import (
15
get_capture_program_passes,
16
)
@@ -29,7 +28,9 @@
29
28
30
def get_instance():
31
module = (
32
- AutoModelForImageClassification.from_pretrained("facebook/deit-base-distilled-patch16-224")
+ AutoModelForImageClassification.from_pretrained(
+ "facebook/deit-base-distilled-patch16-224"
33
+ )
34
.eval()
35
.to("cpu")
36
0 commit comments