Skip to content

Commit cd58c77

Browse files
committed
udpate DeepCache/README.md
1 parent 57310a8 commit cd58c77

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

DeepCache/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ self.logger.log("#Params: {:.4f} M".format(nparams/1e6))
6464
self.logger.log("#MACs: {:.4f} G".format(macs/1e9))
6565
exit()
6666
```
67-
You can set `layer_wise` to `True` to see the FLOPs for each module.
67+
You can enable `layer_wise` to `True` to display the FLOPs for each module. If you are using a different branch for caching, you might need to check and update the `prv_f` shape.
6868

6969
* For Stable Diffusion: (insert it in [Line 752](https://github.com/horseee/DeepCache/blob/fb0ec94e046068eceebe185b2f5cada55b11be1e/DeepCache/sd/pipeline_stable_diffusion.py#L752) for SD pipeline)
7070
```python
@@ -85,7 +85,7 @@ print("#Params: {:.4f} M".format(nparams/1e6))
8585
print("#MACs: {:.4f} G".format(macs/1e9))
8686
exit()
8787
```
88-
You can set `layer_wise` to `True` to see the FLOPs for each module.
88+
To view the FLOPs for each module, you can set the `layer_wise` parameter to `True`. Additionally, if you want to see the FLOPs for the partial model executed during the retrieve steps, you can find the results in the second step with `cache_interval` larger than 2.
8989

9090

9191

0 commit comments

Comments
 (0)