fix the conversation and run success;

This commit is contained in:
junsong
2024-12-05 00:18:56 -08:00
parent 81f58d87cd
commit dde1273804
7 changed files with 33 additions and 769 deletions
-2
View File
@@ -16,10 +16,8 @@
# This file is modified from https://github.com/PixArt-alpha/PixArt-sigma
import math
import os
from typing import Optional
import xformers.ops
import torch
import torch.nn as nn
import torch.nn.functional as F