ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Bug fix

Browse source
This commit is contained in:
Daniel Han 2025-05-02 14:06:14 -07:00
commit 9edbe23259
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/_utils.py
View file

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
__version__ = "2025.4.6"
__version__ = "2025.4.7"
__all__ = [
"SUPPORTS_BFLOAT16",

3
unsloth/models/loader_utils.py
View file

@ -110,7 +110,8 @@ def get_model_name(model_name, load_in_4bit = True):
)
# In the rare case, we convert bad model names to other names
# For eg too large dynamic quants or MoEs
if new_model_name.lower() in BAD_MAPPINGS:
if new_model_name is not None and type(new_model_name) is str and \
new_model_name.lower() in BAD_MAPPINGS:
new_model_name = BAD_MAPPINGS[new_model_name.lower()]
if new_model_name is None and model_name.count("/") == 1 and model_name[0].isalnum():