Bläddra i källkod

Append .amd to reported version

Aleksandr Borzunov 2 år sedan
förälder
incheckning
fe3b8d6e66
2 ändrade filer med 2 tillägg och 2 borttagningar
  1. 1 1
      src/petals/server/handler.py
  2. 1 1
      src/petals/server/server.py

+ 1 - 1
src/petals/server/handler.py

@@ -608,7 +608,7 @@ class TransformerConnectionHandler(ConnectionHandler):
 
         backend = self.module_backends[request.uid] if request.uid else next(iter(self.module_backends.values()))
         result = {
-            "version": petals.__version__,
+            "version": petals.__version__ + ".amd",
             "dht_client_mode": self.dht.client_mode,
             CACHE_TOKENS_AVAILABLE: backend.memory_cache.bytes_left // max(backend.cache_bytes_per_token.values()),
         }

+ 1 - 1
src/petals/server/server.py

@@ -238,7 +238,7 @@ class Server:
         self.server_info = ServerInfo(
             state=ServerState.JOINING,
             public_name=public_name,
-            version=petals.__version__,
+            version=petals.__version__ + ".amd",
             adapters=tuple(adapters),
             torch_dtype=str(torch_dtype).replace("torch.", ""),
             quant_type=quant_type.name.lower(),