no evo

app.py

@@ -40,9 +40,9 @@ custom_css = """
 
 
 os.environ["LAYERNORM_TYPE"] = "fast_layernorm"
-os.environ["USE_DEEPSPEED_EVO_ATTTENTION"] = "true"
+os.environ["USE_DEEPSPEED_EVO_ATTTENTION"] = "False"
 # Set environment variable in the script
-os.environ['CUTLASS_PATH'] = './cutlass'
+#os.environ['CUTLASS_PATH'] = './cutlass'
 
 # reps = [
 #     {
@@ -272,7 +272,7 @@ logging.basicConfig(
     filemode="w",
 )
 configs_base["use_deepspeed_evo_attention"] = (
-    os.environ.get("USE_DEEPSPEED_EVO_ATTTENTION", False) == "true"
+    os.environ.get("USE_DEEPSPEED_EVO_ATTTENTION", False) == "False"
 )
 arg_str = "--seeds 101 --dump_dir ./output --input_json_path ./examples/example.json --model.N_cycle 10 --sample_diffusion.N_sample 5 --sample_diffusion.N_step 200 "
 configs = {**configs_base, **{"data": data_configs}, **inference_configs}

configs/configs_base.py

@@ -103,7 +103,7 @@ model_configs = {
     ),  # NOTE: Number of blocks in each activation checkpoint, if None, no checkpointing is performed.
     # switch of kernels
     "use_memory_efficient_kernel": False,
-    "use_deepspeed_evo_attention": True,
+    "use_deepspeed_evo_attention": False,
     "use_flash": False,
     "use_lma": False,
     "use_xformer": False,