Francesco Capuano commited on
Commit
b976944
·
1 Parent(s): 0ea6096

fix: minor

Browse files
lerobot/scripts/server/robot_client.py CHANGED
@@ -102,14 +102,13 @@ class RobotClient:
102
  server_address: Optional[str] = None,
103
  policy_type: str = "act", # "pi0"
104
  pretrained_name_or_path: str = "fracapuano/act_so100_test", # "lerobot/pi0"
105
- policy_device: str = "cuda",
106
  ):
107
  # Use environment variable if server_address is not provided
108
  if server_address is None:
109
  server_address = os.getenv("SERVER_ADDRESS", "localhost:50051")
110
  logger.info(f"No server address provided, using default address: {server_address}")
111
 
112
- server_address = "172.18.131.21:50051"
113
  self.policy_config = TinyPolicyConfig(policy_type, pretrained_name_or_path, policy_device)
114
  self.channel = grpc.insecure_channel(server_address)
115
  self.stub = async_inference_pb2_grpc.AsyncInferenceStub(self.channel)
 
102
  server_address: Optional[str] = None,
103
  policy_type: str = "act", # "pi0"
104
  pretrained_name_or_path: str = "fracapuano/act_so100_test", # "lerobot/pi0"
105
+ policy_device: str = "mps",
106
  ):
107
  # Use environment variable if server_address is not provided
108
  if server_address is None:
109
  server_address = os.getenv("SERVER_ADDRESS", "localhost:50051")
110
  logger.info(f"No server address provided, using default address: {server_address}")
111
 
 
112
  self.policy_config = TinyPolicyConfig(policy_type, pretrained_name_or_path, policy_device)
113
  self.channel = grpc.insecure_channel(server_address)
114
  self.stub = async_inference_pb2_grpc.AsyncInferenceStub(self.channel)