Spaces:
Runtime error
Runtime error
1111111
Browse files
app.py
CHANGED
@@ -12,6 +12,8 @@ processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-mistral-7b-h
|
|
12 |
model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|
13 |
model.to("cuda:0")
|
14 |
|
|
|
|
|
15 |
@spaces.GPU
|
16 |
def bot_streaming(message, history):
|
17 |
print(message)
|
|
|
12 |
model = LlavaNextForConditionalGeneration.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf", torch_dtype=torch.float16, low_cpu_mem_usage=True)
|
13 |
model.to("cuda:0")
|
14 |
|
15 |
+
image = None
|
16 |
+
|
17 |
@spaces.GPU
|
18 |
def bot_streaming(message, history):
|
19 |
print(message)
|