(self, vae, timestep, scale, seed)
| 47 | CATEGORY = "lightricks/LTXV" |
| 48 | |
| 49 | def add_noise(self, vae, timestep, scale, seed): |
| 50 | result = copy(vae) |
| 51 | if hasattr(result, "first_stage_model"): |
| 52 | result.first_stage_model.decode_timestep = timestep |
| 53 | result.first_stage_model.decode_noise_scale = scale |
| 54 | result._decode_timestep = timestep |
| 55 | result.decode_noise_scale = scale |
| 56 | result.seed = seed |
| 57 | return (result,) |
nothing calls this directly
no outgoing calls
no test coverage detected