Skip to content

Commit 6c21695

Browse files
Update the doc with Environment Testing
1 parent 4ac7b85 commit 6c21695

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

docs/01-Intro.md

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,15 @@ It provides a unified interface for various AI tasks including text-to-image, te
1616

1717
Please refer to the [Model Card](./05-Model%20Card.mdx) for more details.
1818

19+
## Environment Testing
20+
21+
This repository has been tested in environments with `1×A100` and `8×A100` GPUs, using `CUDA 12.4, Python 3.10.16`.
22+
23+
- Cog series models typically do not support `FP16` precision (Only `CogVideoX-2B` support); GPUs like the `V100` cannot be fine-tuned properly (Will cause `loss=nan` for example). At a minimum, an `A100` or other GPUs supporting `BF16` precision should be used.
24+
- We have not yet systematically tested the minimum GPU memory requirements for each model. For `LORA(bs=1 with offload)`, a single `A100` GPU is sufficient. For `SFT`, our tests have passed in an `8×A100` environment.
25+
1926
## Roadmap
2027

2128
- [ ] Add support for CogView4 ControlNet model
22-
- [ ] Docker Image for easy deployment
29+
- [ ] Docker Image for easy deployment
30+
- [ ] Embedding Cache to Reduce GPU Memory Usage

0 commit comments

Comments
 (0)