Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,6 @@ class PromptEnhancerV2:
|
|
| 44 |
self.model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 45 |
models_root_path,
|
| 46 |
torch_dtype=dtype,
|
| 47 |
-
attn_implementation="flash_attention_2",
|
| 48 |
device_map=device_map,
|
| 49 |
)
|
| 50 |
self.processor = AutoProcessor.from_pretrained(models_root_path)
|
|
|
|
| 44 |
self.model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 45 |
models_root_path,
|
| 46 |
torch_dtype=dtype,
|
|
|
|
| 47 |
device_map=device_map,
|
| 48 |
)
|
| 49 |
self.processor = AutoProcessor.from_pretrained(models_root_path)
|