Fixup multiple model DS tests (#3131)

Commit

1 year ago

Fixup multiple model DS tests (#3131) * Multiple model multi GPU fixed, different issues than torch * Fix multiple-model issues