Create FlowMatchEulerDiscreteScheduler config.
(output_path: str, shift: float)
| 156 | |
| 157 | |
| 158 | def create_scheduler_config(output_path: str, shift: float): |
| 159 | """Create FlowMatchEulerDiscreteScheduler config.""" |
| 160 | |
| 161 | scheduler_config = {"_class_name": "FlowMatchEulerDiscreteScheduler", "num_train_timesteps": 1000, "shift": shift} |
| 162 | |
| 163 | scheduler_path = os.path.join(output_path, "scheduler") |
| 164 | os.makedirs(scheduler_path, exist_ok=True) |
| 165 | |
| 166 | with open(os.path.join(scheduler_path, "scheduler_config.json"), "w") as f: |
| 167 | json.dump(scheduler_config, f, indent=2) |
| 168 | |
| 169 | print("✓ Created scheduler config") |
| 170 | |
| 171 | |
| 172 | def download_and_save_vae(vae_type: str, output_path: str): |
no outgoing calls
no test coverage detected
searching dependent graphs…