Pin FAv2 (#1656)

mvpatel2000 · web-flow · commit 0b3a028a6397 · 2024-11-14T13:41:26.000-05:00
diff --git a/setup.py b/setup.py
@@ -104,7 +104,7 @@
 
 # Flash 2 group kept for backwards compatibility
 extra_deps['gpu-flash2'] = [
-    'flash-attn>=2.6.3,<3',
+    'flash-attn==2.6.3',
 ]
 
 extra_deps['gpu'] = copy.deepcopy(extra_deps['gpu-flash2'])

Original file line number	Diff line number	Diff line change
`@@ -104,7 +104,7 @@`
`104`	`104`
`105`	`105`	`# Flash 2 group kept for backwards compatibility`
`106`	`106`	`extra_deps['gpu-flash2'] = [`
`107`		`- 'flash-attn>=2.6.3,<3',`
	`107`	`+ 'flash-attn==2.6.3',`
`108`	`108`	`]`
`109`	`109`
`110`	`110`	`extra_deps['gpu'] = copy.deepcopy(extra_deps['gpu-flash2'])`