Skip to content

Fix skipped inference tests#4336

Merged
mrwyattii merged 2 commits intomasterfrom
mrwyattii/fix-inference-skipped-tests
Sep 15, 2023
Merged

Fix skipped inference tests#4336
mrwyattii merged 2 commits intomasterfrom
mrwyattii/fix-inference-skipped-tests

Conversation

@mrwyattii
Copy link
Contributor

HfApi().list_models() returns a generator. We were using it twice in our setup of the creation of a model-task tuples for the inference tests. This means that the first time we used it, we got the full list. The second time we used it, we got an empty list. Looks like we have been skipping many tests for a few months now!

@loadams

Copy link
Collaborator

@loadams loadams left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Out of curiosity, how much time does this add back to the unit tests?

@mrwyattii mrwyattii enabled auto-merge September 14, 2023 23:12
@mrwyattii mrwyattii added this pull request to the merge queue Sep 14, 2023
Merged via the queue into master with commit 8c7f7fd Sep 15, 2023
CurryRice233 added a commit to CurryRice233/DeepSpeed that referenced this pull request Sep 15, 2023
* origin/master: (48 commits)
  Fix autotune to support Triton 2.1  (deepspeedai#4340)
  Fix skipped inference tests (deepspeedai#4336)
  Suppress noise (deepspeedai#4310)
  Fix a bug in the implementation of dequantization for inference (deepspeedai#3433)
  DS-Chat BLOOM: Fix Attention mask (deepspeedai#4338)
  clear redundant timers (deepspeedai#4308)
  Add release version checking (deepspeedai#4328)
  Fix Zero3 contiguous grads, reduce scatter false  accuracy issue (deepspeedai#4321)
  Clean up modeling code (deepspeedai#4320)
  Handle empty parameter groups (deepspeedai#4277)
  Update README.md (deepspeedai#4316)
  README update (deepspeedai#4303)
  Update release and bump patch versioning flow (deepspeedai#4286)
  added a bert-model check for triton (deepspeedai#4266)
  ZeRO-Inference v2 release
  bump to 0.10.4
  Update index.md (deepspeedai#4297)
  fix user args parsing of string with spaces on runner (deepspeedai#4265)
  ZeRO-Inference refresh (deepspeedai#4197)
  AMD Kernel Compatibility Fixes (deepspeedai#3180)
  ...
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants