Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -584,7 +584,6 @@ model_path = DEFAULT_MODEL
|
|
584 |
logger.info(f"Loading model: {model_path}")
|
585 |
# Check if it's a Qwen model
|
586 |
if "qwenguard" in model_path.lower():
|
587 |
-
@spaces.GPU
|
588 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
589 |
model_path,
|
590 |
torch_dtype="auto",
|
@@ -595,7 +594,6 @@ if "qwenguard" in model_path.lower():
|
|
595 |
|
596 |
# Otherwise assume it's a LlavaGuard model
|
597 |
else:
|
598 |
-
@spaces.GPU
|
599 |
model = LlavaOnevisionForConditionalGeneration.from_pretrained(
|
600 |
model_path,
|
601 |
torch_dtype="auto",
|
|
|
584 |
logger.info(f"Loading model: {model_path}")
|
585 |
# Check if it's a Qwen model
|
586 |
if "qwenguard" in model_path.lower():
|
|
|
587 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
588 |
model_path,
|
589 |
torch_dtype="auto",
|
|
|
594 |
|
595 |
# Otherwise assume it's a LlavaGuard model
|
596 |
else:
|
|
|
597 |
model = LlavaOnevisionForConditionalGeneration.from_pretrained(
|
598 |
model_path,
|
599 |
torch_dtype="auto",
|