Home Explore Help
Register Sign In
AIForce
/
petals
mirror of https://github.com/bigscience-workshop/petals.git
5
0
Fork 0
Files Issues 0 Wiki
Browse Source

typo

justheuristic 3 years ago
parent
4c92d8636f
commit
75fdd9d2cd
1 changed files with 1 additions and 1 deletions
Split View Show Diff Stats
  1. 1 1
      src/client/remote_sequential.py

+ 1 - 1
src/client/remote_sequential.py
View File 

@@ -74,7 +74,7 @@ class RemoteSequential(nn.Sequential):
 
 
     def inference_session(self) -> RemoteSequentialInferenceSession:
 
         self.remote_sequence_info.update_()
 
-        return RemoteSequentialInferenceSession(self.remote_sequence_info)
 
+        return RemoteSequentialInferenceSession(self.remote_sequence_info, self.p2p)
 
 
 
 class RemoteSequentialInferenceSession: