"vscode:/vscode.git/clone" did not exist on "24974b75f8bb4f0a72ac610c589d9d09b3a7f4ec"
Commit a4b666ca authored by gushiqiao's avatar gushiqiao
Browse files

Fix bugs

parent 06f9891a
......@@ -1022,4 +1022,4 @@ if __name__ == "__main__":
model_path = args.model_path
model_cls = args.model_cls
main()
\ No newline at end of file
main()
......@@ -1022,4 +1022,4 @@ if __name__ == "__main__":
model_path = args.model_path
model_cls = args.model_cls
main()
\ No newline at end of file
main()
......@@ -360,4 +360,4 @@ class MemoryBuffer:
self.insertion_index = 0
self.used_mem = 0
torch.cuda.empty_cache()
gc.collect()
\ No newline at end of file
gc.collect()
......@@ -503,4 +503,4 @@ class WanTransformerInfer(BaseTransformerInfer):
if self.clean_cuda_cache:
del y, c_gate_msa
torch.cuda.empty_cache()
return x
\ No newline at end of file
return x
......@@ -169,4 +169,4 @@ def sinusoidal_embedding_1d(dim, position):
x = torch.cat([torch.cos(sinusoid), torch.sin(sinusoid)], dim=1)
if GET_DTYPE() == "BF16":
x = x.to(torch.bfloat16)
return x
\ No newline at end of file
return x
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment