Spaces:
Runtime error
Runtime error
Commit
•
42b4c66
1
Parent(s):
b61da51
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,6 +13,7 @@ from llava.model.builder import load_pretrained_model
|
|
| 13 |
from llava.mm_utils import tokenizer_image_token
|
| 14 |
from transformers.generation.streamers import TextIteratorStreamer
|
| 15 |
import spaces
|
|
|
|
| 16 |
|
| 17 |
|
| 18 |
if torch.cuda.is_available():
|
|
|
|
| 13 |
from llava.mm_utils import tokenizer_image_token
|
| 14 |
from transformers.generation.streamers import TextIteratorStreamer
|
| 15 |
import spaces
|
| 16 |
+
import torch
|
| 17 |
|
| 18 |
|
| 19 |
if torch.cuda.is_available():
|