update reamde

README.md

@@ -119,7 +119,7 @@ with torch.inference_mode():
     
 
 peak_memory_allocated = torch.cuda.max_memory_allocated()
-print(f"Memory Peak: {peak_memory_allocated / (1024**3):.2f} GB") # 转换为GB
+print(f"Memory Peak: {peak_memory_allocated / (1024**3):.2f} GB")
 print(response)
 ```