"references/vscode:/vscode.git/clone" did not exist on "e04de77cbe59effb2fbdb3574e630efb3e194b18"
Commit d2bb2fea authored by gushiqiao's avatar gushiqiao
Browse files

Fix logger bug

parent d1d55ad0
...@@ -29,8 +29,8 @@ Refer to [Quick Start Guide](../getting_started/quickstart.md) to install enviro ...@@ -29,8 +29,8 @@ Refer to [Quick Start Guide](../getting_started/quickstart.md) to install enviro
#### One-Click Gradio Startup (Recommended) #### One-Click Gradio Startup (Recommended)
**📦 Download Software Package** **📦 Download Software Package**
- [Baidu Cloud](https://pan.baidu.com/s/1lLb2LYl8kbmjluaM9-jCjg?pwd=ugd4) - [Baidu Cloud](https://pan.baidu.com/s/14bMAbOuFFSj8yzt4Zdmyqw)
- [Quark Cloud](https://pan.quark.cn/s/d0b8be4bc619) - [Quark Cloud](https://pan.quark.cn/s/97b4ad6bc76c)
**📁 Directory Structure** **📁 Directory Structure**
After extraction, ensure the directory structure is as follows: After extraction, ensure the directory structure is as follows:
......
...@@ -28,8 +28,8 @@ ...@@ -28,8 +28,8 @@
#### 一键启动Gradio(推荐) #### 一键启动Gradio(推荐)
**📦 下载软件包** **📦 下载软件包**
- [百度云](https://pan.baidu.com/s/1lLb2LYl8kbmjluaM9-jCjg?pwd=ugd4) - [百度云](https://pan.baidu.com/s/14bMAbOuFFSj8yzt4Zdmyqw)
- [夸克网盘](https://pan.quark.cn/s/d0b8be4bc619) - [夸克网盘](https://pan.quark.cn/s/97b4ad6bc76c)
**📁 目录结构** **📁 目录结构**
解压后,确保目录结构如下: 解压后,确保目录结构如下:
......
from loguru import logger
try: try:
import flash_attn import flash_attn
from flash_attn.flash_attn_interface import flash_attn_varlen_func from flash_attn.flash_attn_interface import flash_attn_varlen_func
......
...@@ -4,6 +4,7 @@ from lightx2v.utils.registry_factory import ATTN_WEIGHT_REGISTER ...@@ -4,6 +4,7 @@ from lightx2v.utils.registry_factory import ATTN_WEIGHT_REGISTER
import torch.distributed as dist import torch.distributed as dist
from .utils.ring_comm import RingComm from .utils.ring_comm import RingComm
import torch.nn.functional as F import torch.nn.functional as F
from loguru import logger
try: try:
import flash_attn import flash_attn
......
import torch import torch
from .template import AttnWeightTemplate from .template import AttnWeightTemplate
from lightx2v.utils.registry_factory import ATTN_WEIGHT_REGISTER from lightx2v.utils.registry_factory import ATTN_WEIGHT_REGISTER
from loguru import logger
if torch.cuda.get_device_capability(0) == (8, 9): if torch.cuda.get_device_capability(0) == (8, 9):
try: try:
......
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