Upload model

unitary compilation, qc target, 3 qubits, 12 gates max

config.yaml

@@ -0,0 +1,77 @@
+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
+

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:55170daeba77c78a94b99472fa3c207a04532b192fe5d022b04fe8179f119d1c
+size 28839173