mirror of
https://github.com/exo-explore/exo.git
synced 2026-01-26 06:49:37 -05:00
Compare commits
1 Commits
ciaran/ima
...
leo/fix-ba
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9a92eea656 |
@@ -337,6 +337,7 @@ def tensor_auto_parallel(
|
||||
if hasattr(model, "shard") and not isinstance(model, GptOssModel):
|
||||
try:
|
||||
model.shard(group) # type: ignore
|
||||
mx.eval(model.parameters())
|
||||
return patch_tensor_model(model)
|
||||
except (AttributeError, TypeError, NameError):
|
||||
pass
|
||||
|
||||
Reference in New Issue
Block a user