mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-03-10 05:35:59 +08:00
@@ -10,7 +10,7 @@ try:
|
||||
except ImportError:
|
||||
print("Please upgrade `transformers`.")
|
||||
|
||||
from llmtuner.extras.packages import is_flash_attn2_available
|
||||
from ..packages import is_flash_attn2_available
|
||||
|
||||
|
||||
if is_flash_attn2_available():
|
||||
|
||||
Reference in New Issue
Block a user