Spaces:
Paused
Paused
IceClear
commited on
Commit
·
341bd76
1
Parent(s):
f9ea190
update
Browse files
projects/video_diffusion_sr/infer.py
CHANGED
|
@@ -68,7 +68,7 @@ class VideoDiffusionInfer():
|
|
| 68 |
return cond
|
| 69 |
raise NotImplementedError
|
| 70 |
|
| 71 |
-
@log_on_entry
|
| 72 |
@log_runtime
|
| 73 |
def configure_dit_model(self, device="cpu", checkpoint=None):
|
| 74 |
# Load dit checkpoint.
|
|
@@ -99,7 +99,7 @@ class VideoDiffusionInfer():
|
|
| 99 |
num_params = sum(p.numel() for p in self.dit.parameters() if p.requires_grad)
|
| 100 |
print(f"DiT trainable parameters: {num_params:,}")
|
| 101 |
|
| 102 |
-
@log_on_entry
|
| 103 |
@log_runtime
|
| 104 |
def configure_vae_model(self):
|
| 105 |
# Create vae model.
|
|
|
|
| 68 |
return cond
|
| 69 |
raise NotImplementedError
|
| 70 |
|
| 71 |
+
# @log_on_entry
|
| 72 |
@log_runtime
|
| 73 |
def configure_dit_model(self, device="cpu", checkpoint=None):
|
| 74 |
# Load dit checkpoint.
|
|
|
|
| 99 |
num_params = sum(p.numel() for p in self.dit.parameters() if p.requires_grad)
|
| 100 |
print(f"DiT trainable parameters: {num_params:,}")
|
| 101 |
|
| 102 |
+
# @log_on_entry
|
| 103 |
@log_runtime
|
| 104 |
def configure_vae_model(self):
|
| 105 |
# Create vae model.
|