Spaces:

toshi456
/

llava-jp-1.3b-v1.1

Sleeping

toshi456 commited on Apr 24

Commit

586d5b2

•

1 Parent(s): 99a04ab

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import gradio as gr
 import torch
 import transformers
@@ -31,7 +32,7 @@ model.eval()
 conv_mode = "v1"
-@torch.inference_mode()
 def inference_fn(
     image,
     prompt,

+import spaces
 import gradio as gr
 import torch
 import transformers
 conv_mode = "v1"
+@spaces.GPU
 def inference_fn(
     image,
     prompt,