Floki00's picture
Create config.yaml
fc1456f verified
target: genQC.pipeline.multimodal_diffusion_pipeline.MultimodalDiffusionPipeline_ParametrizedCompilation
params:
scheduler:
target: genQC.scheduler.scheduler_ddim.DDIMScheduler
params:
device: cpu
num_train_timesteps: 1000
beta_start: 0.001
beta_end: 0.012
beta_schedule: "path:beta_schedule_learned_linear.safetensors"
input_perturbation: 0.0
prediction_type: v-type
scheduler_w:
target: genQC.scheduler.scheduler_ddim.DDIMScheduler
params:
device: cpu
num_train_timesteps: 1000
beta_start: 0.0002
beta_end: 0.024
beta_schedule: cos_alpha4
input_perturbation: 0.0
prediction_type: v-type
model:
target: genQC.models.transformers.cirdit_multimodal.UnitaryCLIPPartialNoiseCompilationCirDiT
save_type: safetensors
params:
clr_dim: 16
ch_packing: 384
ch_core: 1024
c_emb_size: 640
t_emb_size: 512
depth_packing: 6
depth_core: 12
num_heads_packing: 12
num_heads_core: 16
dropout: 0.0
p_rope: 0.9
base_rope: 1000.0
unitary_encoder_config:
target: genQC.models.clip.unitary_clip.UnitaryTextEncoder
params:
text_embed_ch: 512
text_encoding_ch: 128
text_attn_num_heads: 8
text_attn_depth: 1
unitary_encoding_ch: 40
unitary_downscale_factor: 4
main_num_heads: 16
main_depth: 8
use_rope: false
p_rope: 0.9
base_rope: 1000.0
dropout: 0.1
text_encoder:
target: genQC.models.clip.frozen_open_clip.CachedFrozenOpenCLIPEmbedder
params:
arch: ViT-B-32
version: datacomp_xl_s13b_b90k
#device: cpu
max_length: 37
freeze: true
layer: penultimate
enable_cache_token_limit: true
embedder:
target: genQC.models.embedding.rotational_preset_embedder.RotationalMultimodialPresetEmbedderTiny
save_type: safetensors
params:
clr_dim: 13
num_clrs: 12
params_dim: 3
num_params_per_clr: 1
zero_sum_space: true
explicit_node_type_embeddings: false
channel_last: true
parametrized_tokens:
- 5
- 6
- 7
- 8
unique_class_values:
- -3
- -2
- 0
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
device: cpu
enable_guidance_train: true
guidance_train_p: 0.1
cached_text_enc: true
add_config:
dataset:
params:
pad_constant: 9
num_of_qubits: 5
min_gates: 4
max_gates: 32
max_params: 1
gate_pool:
- h
- cx
- ccx
- swap
- rx
- ry
- rz
- cp