Skip to content

Commit 3a4280c

Browse files
committed
Merge remote-tracking branch 'upstream/main' into add_afmoe_model
2 parents 1ae79d2 + 1fd63dd commit 3a4280c

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

52 files changed

+5212
-522
lines changed

.github/workflows/benchmark.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@ jobs:
4040
run: python3 -m pip install -r benchmark_v2/requirements.txt kernels
4141

4242
- name: Reinstall transformers in edit mode (remove the one installed during docker image build)
43-
working-directory: /transformers
4443
run: python3 -m pip uninstall -y transformers && python3 -m pip install -e ".[torch]"
4544

4645
- name: Run benchmark

benchmark_v2/framework/benchmark_runner.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,6 @@ def flush_memory():
117117
# Clear CUDA cache
118118
if torch.cuda.is_available():
119119
torch.cuda.empty_cache()
120-
torch.cuda.reset_max_memory_allocated()
121-
torch.cuda.reset_peak_memory_stats()
122120
torch.cuda.synchronize()
123121
gc.collect()
124122

docs/source/en/_toctree.yml

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1010,6 +1010,8 @@
10101010
title: AltCLIP
10111011
- local: model_doc/aria
10121012
title: Aria
1013+
- local: model_doc/audioflamingo3
1014+
title: AudioFlamingo3
10131015
- local: model_doc/aya_vision
10141016
title: AyaVision
10151017
- local: model_doc/blip

0 commit comments

Comments
 (0)