Shaoan commited on
Commit
ad56805
·
verified ·
1 Parent(s): f15853b

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. aligner.py +1 -1
  2. requirements.txt +16 -13
aligner.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
  from torch import nn
3
- from refiner import Qwen2Connector
4
 
5
  import torch
6
  import torch.nn as nn
 
1
  import torch
2
  from torch import nn
3
+ #from refiner import Qwen2Connector
4
 
5
  import torch
6
  import torch.nn as nn
requirements.txt CHANGED
@@ -1,13 +1,16 @@
1
- torch==2.0.1
2
- torchvision==0.15.2
3
- gradio==4.44.0
4
- diffusers==0.27.2
5
- transformers==4.40.0
6
- safetensors==0.4.2
7
- accelerate==0.27.2
8
- peft==0.8.2
9
- Pillow==10.2.0
10
- huggingface_hub==0.20.3
11
- einops==0.7.0
12
- sentencepiece==0.1.99
13
- protobuf==4.25.3
 
 
 
 
1
+ accelerate==1.10.1
2
+ diffusers==0.35.2
3
+ einops==0.8.1
4
+ gradio==3.50.2
5
+ gradio_client==0.6.1
6
+ huggingface-hub==0.35.3
7
+ numpy==1.26.4
8
+ peft==0.17.1
9
+ pillow==10.4.0
10
+ protobuf==6.32.1
11
+ pytorch-triton-rocm==3.4.0
12
+ safetensors==0.6.2
13
+ sentencepiece==0.2.1
14
+ torch==2.8.0+rocm6.4
15
+ torchvision==0.23.0+rocm6.4
16
+ transformers==4.57.1