Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions QEfficient/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,14 +24,16 @@
QEFFAutoModelForCausalLM,
QEFFAutoModelForCTC,
QEFFAutoModelForImageTextToText,
QEFFAutoModelForMultimodalLM,
QEFFAutoModelForSequenceClassification,
QEFFAutoModelForSpeechSeq2Seq,
QEFFCommonLoader,
)
from QEfficient.compile.compile_helper import compile
from QEfficient.diffusers.pipelines.flux.pipeline_flux import QEffFluxPipeline
from QEfficient.diffusers.pipelines.wan.pipeline_wan import QEffWanPipeline
from QEfficient.diffusers.pipelines.wan.pipeline_wan_i2v import QEffWanImageToVideoPipeline

# from QEfficient.diffusers.pipelines.flux.pipeline_flux import QEffFluxPipeline
# from QEfficient.diffusers.pipelines.wan.pipeline_wan import QEffWanPipeline
# from QEfficient.diffusers.pipelines.wan.pipeline_wan_i2v import QEffWanImageToVideoPipeline
from QEfficient.exporter.export_hf_to_cloud_ai_100 import qualcomm_efficient_converter
from QEfficient.generation.text_generation_inference import cloud_ai_100_exec_kv
from QEfficient.peft import QEffAutoPeftModelForCausalLM
Expand All @@ -55,6 +57,7 @@
"QEFFAutoModelForCTC",
"QEffAutoPeftModelForCausalLM",
"QEFFAutoModelForImageTextToText",
"QEFFAutoModelForMultimodalLM",
"QEFFAutoModelForSequenceClassification",
"QEFFAutoModelForSpeechSeq2Seq",
"QEFFCommonLoader",
Expand Down
1 change: 1 addition & 0 deletions QEfficient/base/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
QEFFAutoModelForCausalLM,
QEFFAutoModelForCTC,
QEFFAutoModelForImageTextToText,
QEFFAutoModelForMultimodalLM,
QEFFAutoModelForSequenceClassification,
QEFFAutoModelForSpeechSeq2Seq,
)
420 changes: 208 additions & 212 deletions QEfficient/transformers/cache_utils.py

Large diffs are not rendered by default.

Loading
Loading