File size: 1,305 Bytes
370f342
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# Model Files Information

## Available Files
- normalization_stats.pt: Preprocessing statistics for ESM-2 embeddings

## Missing Files (Too Large for Hugging Face)
The following model files are too large for Hugging Face upload (>100MB limit):

### Large Model Files (Not Included)
- flowamp_demo_checkpoint.pth (~1.5GB): Complete model checkpoint
- compressor_demo.pth (~315MB): Compressor weights
- decompressor_demo.pth (~158MB): Decompressor weights  
- flow_model_demo.pth (~54MB): Flow model weights
- apex/trained_models/* (~1GB total): Pre-trained Apex models

### How to Get Model Files
1. **Train Your Own**: Use the provided training scripts to train the model
2. **Contact Author**: Request model files directly from the author
3. **Alternative Storage**: Model files may be available on other platforms

### Training Instructions
1. Run the training scripts to generate your own model checkpoints
2. Use amp_flow_training_single_gpu_full_data.py for single GPU training
3. Use amp_flow_training_multi_gpu.py for multi-GPU training
4. Models will be saved automatically during training

### Quick Start
1. Install dependencies: pip install -r requirements.txt
2. Run usage_example.py to verify installation
3. Train the model using provided scripts
4. Use generate_amps.py for AMP generation