Minor tweak to test

stablecog · Sep 7, 2024 · d33207b · d33207b
1 parent 42129d2
commit d33207b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/flux1/generate.py b/src/flux1/generate.py
@@ -15,11 +15,11 @@
 def generate(
     props: GenerateFunctionProps[Flux1PipeObject],
 ) -> List[GenerateOutput]:
-    # Props -------------------------------------
+    # Props ------------------------------------
     input = props.input
     pipe_object = props.pipe_object
     model_name = props.model_name
-    # -------------------------------------------
+    # ------------------------------------------
 
     inference_start = time.time()
     log_gpu_memory(message="Before inference")