Skip to content

Commit e4d8d5e

Browse files
committed
corretly handle model property
1 parent f296177 commit e4d8d5e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/lightning/fabric/strategies/deepspeed.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def distributed_sampler_kwargs(self) -> dict[str, int]:
294294

295295
@property
296296
def model(self) -> "DeepSpeedEngine":
297-
return self.deepspeed_impl.model
297+
return self.deepspeed_impl._deepspeed_engine
298298

299299
@override
300300
def setup_module_and_optimizers(

0 commit comments

Comments
 (0)