CollaGen: Collaborative Learning Tool for Generative Model
Project description
CollaGen Framework
Collaborative Generation Framework for Generative Model
System Environments
Python Dependent Packages
Packagename | Version | Note |
---|---|---|
loguru | 0.7.2 | == to latest |
pyyaml | 6.0.1 | == to latest |
wandb | 0.16.4 | == to latest |
torch | 2.2.1 | == to latest |
torchvision | 0.17.1 | == to latest |
datasets | 2.18.0 | == to latest, FLock simulation |
peft | 0.9.0 | == to latest, FLock simulation |
flask | 3.0.2 | == to latest, FLock simulation |
trl | 0.7.11 | == to latest, FLock simulation |
bitsandbytes | 0.43.0 | == to latest, FLock simulation |
Acknowledgments
We thank the following repos providing helpful components/functions in our work.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
collagenai-0.0.1.tar.gz
(11.2 kB
view hashes)
Built Distribution
collagenai-0.0.1-py3-none-any.whl
(13.3 kB
view hashes)
Close
Hashes for collagenai-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 249e39c430f625d41facd335659c7169607bb7ed3a1d6ceeafd7437531f0b8be |
|
MD5 | 174a91b452d6479a84296de6d1ae462a |
|
BLAKE2b-256 | 6204329c3d9634134c85acecb2ac6f28076f2221d4327cfe67883f94c9be0451 |