modifying ray.py

Author: webcoderz

modules/api/ray.py

@@ -11,33 +11,27 @@
 #ray.init("ray://localhost:10001")
 
 
-ray_head_address = os.environ.get("RAY_HEAD_ADDRESS")
-print("RAY_HEAD_ADDRESS:", ray_head_address)
-
-if ray_head_address:
-    ray.init(address=ray_head_address)
+if "RAY_DOCKER" in os.environ:
+    ray.init(
+        dashboard_host=os.environ.get("RAY_DASHBOARD_HOST", "0.0.0.0"), 
+        dashboard_port=int(os.environ.get("RAY_DASHBOARD_PORT", 8000))
+        )
 else:
     ray.init()
 
 
 def ray_only():
     serve.shutdown()
-    if "RAY_DOCKER" in os.environ:
-        print("starting ray in docker")
-        serve.start(
-            detached=True,
-            http_options={
-                        "host": os.environ.get("RAY_IP", "0.0.0.0"), 
-                        "port": int(os.environ.get("RAY_PORT", 8000))
-                        }
-        )
-    else:
-        serve.start(
+    
+    print("starting ray")
+    serve.start(
+            
             http_options={
                         "host": os.environ.get("RAY_IP", "0.0.0.0"), 
                         "port": int(os.environ.get("RAY_PORT", 8000))
                         }
         )
+
     print(f"Starting Raypi on port {os.environ.get('RAY_PORT', 8000)}")
     serve.run(Raypi.bind(), port=int(os.environ.get("RAY_PORT", 8000)), route_prefix="/sdapi/v1")  #route_prefix="/sdapi/v1" # Call the launch_ray method to get the FastAPI app
     print("Done setting up replicas! Now accepting requests...")

webui-user.sh

@@ -52,5 +52,7 @@
 #export RAY_MAX_REPLICAS=2
 #export RAY_PORT=7575
 #export RAY_IP="0.0.0.0"
-#export RAY_HEAD_ADDRESS='ray://ray-head:10001'
+#export RAY_ADDRESS='ray://ray-head:10001'
 #export RAY_DOCKER="True"
+#export RAY_DASHBOARD_PORT=8265
+#export RAY_DASHBOARD_HOST="0.0.0.0"