Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
File size: 279 Bytes
5ea344f 2201358 5ea344f 2201358 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import gradio as gr
import spaces
import transformers_gradio
demo = gr.load(name="allenai/Llama-3.1-Tulu-3-8B", src=transformers_gradio.registry)
demo.fn = spaces.GPU()(demo.fn)
for fn in demo.fns.values():
fn.api_name = False
if __name__ == "__main__":
demo.launch()
|