(safetensors_path)
| 110 | |
| 111 | |
| 112 | def is_dev(safetensors_path): |
| 113 | with safe_open(safetensors_path, framework='pt', device='cpu') as f: |
| 114 | for key in f.keys(): |
| 115 | if key.startswith('guidance_in'): |
| 116 | return True |
| 117 | return False |
| 118 | |
| 119 | |
| 120 | def time_shift(mu: float, sigma: float, t: torch.Tensor): |