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