Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -86,6 +86,7 @@ def get_conv_log_filename():
|
|
86 |
|
87 |
def get_model_list():
|
88 |
models = [
|
|
|
89 |
'LukasHug/LlavaGuard-7B-hf',
|
90 |
'LukasHug/LlavaGuard-13B-hf',
|
91 |
'LukasHug/LlavaGuard-34B-hf', ][:1]
|
@@ -498,11 +499,12 @@ Set the environment variable `model` to change the model:
|
|
498 |
|
499 |
|
500 |
models = [
|
|
|
501 |
'LukasHug/LlavaGuard-7B-hf',
|
502 |
'LukasHug/LlavaGuard-13B-hf',
|
503 |
'LukasHug/LlavaGuard-34B-hf', ]
|
504 |
bits = int(os.getenv("bits", 16))
|
505 |
-
model = os.getenv("model", models[
|
506 |
available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
|
507 |
model_path, model_name = model, model.split("/")[0]
|
508 |
if api_key:
|
|
|
86 |
|
87 |
def get_model_list():
|
88 |
models = [
|
89 |
+
'AIML-TUDA/LlavaGuard-7B',
|
90 |
'LukasHug/LlavaGuard-7B-hf',
|
91 |
'LukasHug/LlavaGuard-13B-hf',
|
92 |
'LukasHug/LlavaGuard-34B-hf', ][:1]
|
|
|
499 |
|
500 |
|
501 |
models = [
|
502 |
+
'AIML-TUDA/LlavaGuard-7B',
|
503 |
'LukasHug/LlavaGuard-7B-hf',
|
504 |
'LukasHug/LlavaGuard-13B-hf',
|
505 |
'LukasHug/LlavaGuard-34B-hf', ]
|
506 |
bits = int(os.getenv("bits", 16))
|
507 |
+
model = os.getenv("model", models[0])
|
508 |
available_devices = os.getenv("CUDA_VISIBLE_DEVICES", "0")
|
509 |
model_path, model_name = model, model.split("/")[0]
|
510 |
if api_key:
|