vovahimself commited on
Commit
e789c49
1 Parent(s): f242ec5

transformers' jukebox

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. requirements.txt +3 -0
app.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import gradio as gr
2
 
3
  def greet(name):
 
1
+ from transformers import JukeboxModel , JukeboxTokenizer
2
+ from transformers.models.jukebox import convert_jukebox
3
+
4
  import gradio as gr
5
 
6
  def greet(name):
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ git+https://github.com/ArthurZucker/transformers.git@jukebox
2
+ accelerate
3
+ bitsandbytes==0.31.8