Add missing exception
This commit is contained in:
parent
3345da2ea4
commit
fbac6d21ca
1 changed files with 4 additions and 0 deletions
|
|
@ -1,4 +1,5 @@
|
|||
import os
|
||||
import traceback
|
||||
from pathlib import Path
|
||||
from typing import Any, Dict, Optional, Union
|
||||
|
||||
|
|
@ -21,6 +22,9 @@ except ModuleNotFoundError:
|
|||
'https://github.com/Dao-AILab/flash-attention#installation-and-features'
|
||||
)
|
||||
pass
|
||||
except Exception:
|
||||
logger.warning('Failed to load flash-attention due to the following error:\n')
|
||||
traceback.print_exc()
|
||||
|
||||
|
||||
class Exllamav2HF(PreTrainedModel):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue