Commit 8cd36771 authored by gushiqiao's avatar gushiqiao Committed by GitHub
Browse files

Fix logger bug

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