Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files
app.py
CHANGED
@@ -11,7 +11,6 @@ model_path = 'GD-ML/UniVG-R1'
|
|
11 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
12 |
model_path,
|
13 |
torch_dtype=torch.bfloat16,
|
14 |
-
attn_implementation="flash_attention_2",
|
15 |
device_map="cuda:0",
|
16 |
)
|
17 |
processor = AutoProcessor.from_pretrained(model_path, max_pixels=401408)
|
|
|
11 |
model = Qwen2VLForConditionalGeneration.from_pretrained(
|
12 |
model_path,
|
13 |
torch_dtype=torch.bfloat16,
|
|
|
14 |
device_map="cuda:0",
|
15 |
)
|
16 |
processor = AutoProcessor.from_pretrained(model_path, max_pixels=401408)
|