(self,
sample,
timestamp,
encoder_hidden_states,
return_dict=True,
cross_attention_kwargs=None,
timestep_cond=None,
added_cond_kwargs=None)
| 64 | self.cuda_graph_created = True |
| 65 | |
| 66 | def _forward(self, |
| 67 | sample, |
| 68 | timestamp, |
| 69 | encoder_hidden_states, |
| 70 | return_dict=True, |
| 71 | cross_attention_kwargs=None, |
| 72 | timestep_cond=None, |
| 73 | added_cond_kwargs=None): |
| 74 | if cross_attention_kwargs: |
| 75 | return self.unet(sample, |
| 76 | timestamp, |
| 77 | encoder_hidden_states, |
| 78 | return_dict, |
| 79 | cross_attention_kwargs=cross_attention_kwargs) |
| 80 | else: |
| 81 | return self.unet(sample, timestamp, encoder_hidden_states, return_dict) |
no outgoing calls
no test coverage detected