tjl223 commited on
Commit
3d922cc
1 Parent(s): fe841f8

add git ignore

Browse files
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ __pycache__/
LyricGeneratorModel.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
 
3
- from peft import LoraConfig, get_peft_model, PeftModel, PeftConfig
4
 
5
 
6
  class LyricGeneratorModel:
 
1
  from transformers import AutoTokenizer, AutoModelForCausalLM
2
 
3
+ from peft import PeftModel, PeftConfig
4
 
5
 
6
  class LyricGeneratorModel:
__pycache__/ArtistCoherencyModel.cpython-312.pyc DELETED
Binary file (5.43 kB)
 
__pycache__/ArtistCoherencyModel.cpython-39.pyc DELETED
Binary file (3.4 kB)
 
__pycache__/FFNN.cpython-312.pyc DELETED
Binary file (5.49 kB)
 
__pycache__/LyricGeneratorModel.cpython-312.pyc DELETED
Binary file (1.68 kB)
 
requirements.txt CHANGED
@@ -3,4 +3,6 @@ huggingface_hub==0.22.2
3
  torch==2.2.2
4
  numpy==1.26.4
5
  pandas==2.2.2
6
- peft==0.10.0
 
 
 
3
  torch==2.2.2
4
  numpy==1.26.4
5
  pandas==2.2.2
6
+ peft==0.10.0
7
+ bitsandbytes==0.42.0
8
+ accelerate==0.29.3