Fix model loading issues
This commit is contained in:
+1
-1
@@ -1,4 +1,4 @@
|
||||
dtypes = ["float32", "float16"]
|
||||
dtypes = ["float32", "float16", "bfloat16"]
|
||||
|
||||
device_map = "auto"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user