beam_search_visualizer

Runtime error

FlipTip commited on May 16

Commit

042471e

•

1 Parent(s): d838787

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import numpy as np
 import gradio as gr
 import spaces
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3-mini-4k-instruct",

 import numpy as np
 import gradio as gr
 import spaces
+import accelerate
 model = AutoModelForCausalLM.from_pretrained(
     "microsoft/Phi-3-mini-4k-instruct",