File size: 1,614 Bytes
a738617 |
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 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
target: genQC.pipeline.diffusion_pipeline_special.DiffusionPipeline_Compilation
params:
scheduler:
target: genQC.scheduler.scheduler_ddim.DDIMScheduler
params:
device: cpu
num_train_timesteps: 1000
beta_start: 0.0001
beta_end: 0.02
beta_schedule: cos_alpha
input_perturbation: 0.1
eta: 1
model:
target: genQC.models.unet_qc.QC_Compilation_UNet
save_path: null
params:
model_features:
- 128
- 128
- 256
clr_dim: 8
num_clrs: 8
t_emb_size: 256
cond_emb_size: 512
num_heads:
- 8
- 8
- 2
num_res_blocks:
- 2
- 2
- 4
transformer_depths:
- 1
- 2
- 1
unitary_encoder_config:
cond_emb_size: 512
model_features:
- 2
- 32
- 64
- 512
num_heads: 8
transformer_depths:
- 2
- 2
dropout: 0.2
text_encoder:
target: genQC.models.frozen_open_clip.CachedFrozenOpenCLIPEmbedder
save_path: null
params:
arch: ViT-B-32
version: laion2b_s34b_b79k
device: cpu
max_length: 77
freeze: true
layer: penultimate
device: cpu
enable_guidance_train: true
guidance_train_p: 0.1
cached_text_enc: true
add_config:
dataset:
params:
num_of_qubits: 3
min_gates: 2
max_gates: 12
gate_pool:
- h
- cx
- z
- x
- ccx
- swap
pad_constant: 7
|