修改环境变量

Diffusion_test_offload_false.py

@@ -397,17 +397,13 @@ class StableDiffusionMGX():
 
 
 if __name__ == "__main__":
-    os.environ['MIGRAPHX_DISABLE_LAYERNORM_FUSION'] = '1'
-    os.environ['MIGRAPHX_STATIC_SHAPE_MODE'] = '1'
     args = get_args()
 
     sd = StableDiffusionMGX(args.onnx_model_path, args.compiled_model_path,
                             args.fp16, args.force_compile)
     sd.warmup(5)
-    start_time = time.time()
     result = sd.run(args.prompt, args.negative_prompt, args.steps, args.seed,
                     args.scale)
-    print("Run a image cost : {}".format(time.time() - start_time))
 
     print("Convert result to rgb image...")
     image = StableDiffusionMGX.convert_to_rgb_image(result)