refine log

Author: strint

src/diffusers/oneflow_graph_compile_cache.py

@@ -159,11 +159,9 @@ def load_graph(self, path):
             for sub_folder in sub_folders:
                 state_dict = flow.load(sub_folder)
                 cache_order = state_dict["cache_order"]
-                print("===> order", cache_order)
                 graph_dict[cache_order] = state_dict
 
             for order, state_dict in sorted(graph_dict.items()):
-                print("===> load order", order)
                 graph_class_name  = state_dict["graph_class_name"]
                 cache_key = state_dict["cache_key"]
                 if graph_class_name not in self.cache_bucket_:

src/diffusers/utils/__init__.py

@@ -101,6 +101,7 @@ def new_fn(*args, **kwargs):
         print(fn.__name__, " run time ", end_time - start_time)
         print(fn.__name__, " cuda mem", after_used - before_used)
         print("<==", fn.__name__, " finish run")
+        print("")
         return out
 
     return new_fn