Fix

examples/ray/ray_flux_example.py

@@ -53,10 +53,10 @@ def main():
     # output is a list of results from each worker, we take the last one
     for i, image in enumerate(output[-1].images):
         image.save(
-            f"/data/results/{model_name}_result_{i}.png"
+            f"./results/{model_name}_result_{i}.png"
         )
         print(
-            f"image {i} saved to /data/results/{model_name}_result_{i}.png"
+            f"image {i} saved to ./results/{model_name}_result_{i}.png"
         )
 
 

examples/ray/ray_run.sh

@@ -3,7 +3,7 @@ set -x
 export PYTHONPATH=$PWD:$PYTHONPATH
 
 # Select the model type
-export MODEL_TYPE="Sd3"
+export MODEL_TYPE="Flux"
 # Configuration for different model types
 # script, model_id, inference_step
 declare -A MODEL_CONFIGS=(

examples/ray/ray_sd3_example.py

@@ -66,10 +66,10 @@ def main():
     # output is a list of results from each worker, we take the last one
     for i, image in enumerate(output[-1].images):
         image.save(
-            f"/data/results/{model_name}_result_{i}.png"
+            f"./results/{model_name}_result_{i}.png"
         )
         print(
-            f"image {i} saved to /data/results/{model_name}_result_{i}.png"
+            f"image {i} saved to ./results/{model_name}_result_{i}.png"
         )
 
 

xfuser/config/config.py

@@ -192,7 +192,6 @@ class ParallelConfig:
     sp_config: SequenceParallelConfig
     pp_config: PipeFusionParallelConfig
     tp_config: TensorParallelConfig
-    distributed_executor_backend: Optional[str] = None
     world_size: int = 1 # FIXME: remove this
     worker_cls: str = "xfuser.ray.worker.worker.Worker"
 

xfuser/ray/worker/worker.py

@@ -1,4 +1,3 @@
-
 from abc import ABC, abstractmethod
 
 from xfuser.core.distributed import (