spatial-banana / DOWNLOAD.md
Biogod's picture
Release Mix704 and prostate grading adapters
fe57637 verified
|
Raw
History Blame Contribute Delete
4.67 kB

Released Weights

Recommended Adapter

The current recommended generalist release is the Mix704 step-8,800 rank-128 task LoRA:

Field Value
Hugging Face repository Biogod/spatial-banana
Remote path experiments/multimodal/e6_sourcecap32_e125_v1/runs/mix704_step8800/final/pytorch_lora_weights.safetensors
Size 1,699,695,208 bytes
SHA-256 cb21fc1044e7f7721f0242b05581fa386e240eb35da4d9a6ef9cd27ddddfe830
Base model Qwen/Qwen-Image-Edit-2511
Acceleration LoRA lightx2v/Qwen-Image-Edit-2511-Lightning, four-step BF16
Local expected directory runs/patho_banana/multimodal_cell_segmentation/e6_sourcecap32_e125_v1/e6_mix704_sourcecap32_r128_e125_v1.early_stop_recovery_step_008800/resume-state-step-008800/task_lora/

The adapter is not a standalone pipeline. It must be loaded after fusing the four-step Lightning LoRA into the frozen Qwen-Image-Edit-2511 transformer.

Download

Install the repository environment first:

uv sync --extra train --extra data --extra dev

The commands below pin every remote revision, use HF-Mirror without inheriting proxy variables, keep Xet enabled, and place files at the paths expected by the checked-in benchmark configuration.

env -u HTTP_PROXY -u HTTPS_PROXY -u ALL_PROXY \
  -u http_proxy -u https_proxy -u all_proxy \
  HF_ENDPOINT=https://hf-mirror.com \
  HF_XET_HIGH_PERFORMANCE=1 \
  hf download Qwen/Qwen-Image-Edit-2511 \
  --revision 6f3ccc0b56e431dc6a0c2b2039706d7d26f22cb9 \
  --local-dir weights/qwen-image-edit-2511/diffusers \
  --max-workers "$(nproc)"

mkdir -p weights/loras
env -u HTTP_PROXY -u HTTPS_PROXY -u ALL_PROXY \
  -u http_proxy -u https_proxy -u all_proxy \
  HF_ENDPOINT=https://hf-mirror.com \
  HF_XET_HIGH_PERFORMANCE=1 \
  hf repos cp \
  hf://lightx2v/Qwen-Image-Edit-2511-Lightning@d74eba145674fd7e31b949324e148e21e7118abd/Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors \
  weights/loras/Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors

release_lora_dir=runs/patho_banana/multimodal_cell_segmentation/e6_sourcecap32_e125_v1/e6_mix704_sourcecap32_r128_e125_v1.early_stop_recovery_step_008800/resume-state-step-008800/task_lora
mkdir -p "$release_lora_dir"
env -u HTTP_PROXY -u HTTPS_PROXY -u ALL_PROXY \
  -u http_proxy -u https_proxy -u all_proxy \
  HF_ENDPOINT=https://hf-mirror.com \
  HF_XET_HIGH_PERFORMANCE=1 \
  hf repos cp \
  hf://Biogod/spatial-banana/experiments/multimodal/e6_sourcecap32_e125_v1/runs/mix704_step8800/final/pytorch_lora_weights.safetensors \
  "$release_lora_dir/pytorch_lora_weights.safetensors"

No Hugging Face token is needed while the repository remains public. If authentication is required later, expose HF_TOKEN only to the process; do not write it into a config, command-line argument, or committed file.

Verify

sha256sum \
  weights/loras/Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors \
  "$release_lora_dir/pytorch_lora_weights.safetensors"

Expected hashes:

22226e8d05d354bb356627d428809f5afd7819399b077238a2b70a82883a904f  Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors
cb21fc1044e7f7721f0242b05581fa386e240eb35da4d9a6ef9cd27ddddfe830  pytorch_lora_weights.safetensors

Run the repository preflight before inference:

uv run python scripts/patho_banana/run_cellsam_non_he_benchmark.py \
  --config configs/patho_banana/multimodal_cell_segmentation/benchmarks/cellsam_non_he_qwen_current_default_v3.yaml \
  audit --allow-missing-data

The audit intentionally fails when a weight is absent or has the wrong hash. --allow-missing-data relaxes only the dataset requirement. Full benchmark reproduction additionally requires the CellSAM datasets and reference model described by the resolved protocol.

Evidence and Limitations

The development-only prostate grading adapter is archived separately at experiments/biological_applications/pb_e06_prostate_grading_p01_v1/runs/mix704_r128/final/. It is not the recommended generalist segmentation adapter and is not intended for clinical use.

The remaining adapters in Biogod/spatial-banana are retained as development or recovery artifacts. Their presence does not mean that they passed the final scientific gate.