Ericzhang25 commited on
Commit
6fdba93
·
verified ·
1 Parent(s): 03e6843

Update apptainer/README.md

Browse files
Files changed (1) hide show
  1. apptainer/README.md +109 -0
apptainer/README.md CHANGED
@@ -0,0 +1,109 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # CryoZeta Run Scripts
2
+
3
+ This repository includes local wrapper scripts for running CryoZeta through
4
+ Apptainer with externally downloaded model weights.
5
+
6
+ Use:
7
+
8
+ - `run_cryozeta.sh` for standard CryoZeta inference
9
+ - `run_cryozeta_large.sh` for large-complex cycle inference
10
+
11
+
12
+ ## 1. Download weights once
13
+
14
+ Download the Hugging Face model files into a local assets directory:
15
+
16
+ ```bash
17
+ huggingface-cli download KiharaLab/CryoZeta --repo-type model \
18
+ --local-dir /path/to/assets
19
+ ```
20
+
21
+ That directory should contain the model files used by the wrappers, including:
22
+
23
+ - `cryozeta-detection-v0.0.1.safetensors`
24
+ - `cryozeta-v0.0.1.safetensors`
25
+ - `cryozeta-interpolate-v0.0.1.safetensors`
26
+
27
+
28
+ ## 2. Normal inference
29
+
30
+ Use `run_cryozeta.sh` for the standard pipeline:
31
+
32
+ - Stage 1: atom detection
33
+ - Stage 2: structure prediction
34
+ - Stage 3: result combination
35
+
36
+ Basic usage:
37
+
38
+ ```bash
39
+ ./run_cryozeta.sh --gpu ID <input_json> <output_dir> <assets_dir>
40
+ ```
41
+
42
+ Example:
43
+
44
+ ```bash
45
+ ./run_cryozeta.sh \
46
+ --gpu 0 \
47
+ server/3a7c2af60301441c607c5dfc565add50/input.json \
48
+ server/tmp \
49
+ /path/to/assets
50
+ ```
51
+
52
+ Notes:
53
+
54
+ - Edit the `SIF` path near the top of `run_cryozeta.sh` if needed, or pass
55
+ `--sif PATH`.
56
+ - Pass the downloaded assets directory as the third positional argument.
57
+ - Run the script from the repo root so bind mounts and relative paths resolve
58
+ cleanly.
59
+
60
+
61
+ ## 3. Large-complex inference
62
+
63
+ Use `run_cryozeta_large.sh` for large-complex cycle prediction:
64
+
65
+ - detection
66
+ - cycle prediction
67
+ - combine stages
68
+
69
+ Basic usage:
70
+
71
+ ```bash
72
+ ./run_cryozeta_large.sh --sif /path/to/CryoZeta.sif --gpu ID \
73
+ <input_json> <output_dir> <assets_dir>
74
+ ```
75
+
76
+ Example:
77
+
78
+ ```bash
79
+ ./run_cryozeta_large.sh \
80
+ --sif /path/to/CryoZeta.sif \
81
+ --gpu 0 \
82
+ server/8zui/input.json \
83
+ server/tmp \
84
+ /path/to/assets
85
+ ```
86
+
87
+ Optional flags:
88
+
89
+ - `--example SEL` select one entry by index or name from a JSON list
90
+ - `--registration auto|teaser|svd|vesper`
91
+ - `--threshold X`
92
+ - `--n-sample N`
93
+ - `--n-step N`
94
+ - `--n-cycle N`
95
+ - `--skip-detection`
96
+ - `--skip-combine`
97
+
98
+ Path handling:
99
+
100
+ - `run_cryozeta_large.sh` supports both JSON-local relative paths such as
101
+ `60484.map` and repo-root-relative paths such as `server/8zui/60484.map`.
102
+
103
+
104
+ ## 4. Recommended workflow
105
+
106
+ 1. Download weights once with `huggingface-cli`.
107
+ 2. Use `run_cryozeta.sh` for normal inference jobs.
108
+ 3. Use `run_cryozeta_large.sh` for large-complex jobs.
109
+