protectai / modelscan

Protection against Model Serialization Attacks

Home Page:http://modelscan.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Handle whitespaces in HuggingFace URLs

iamfaisalkhan opened this issue · comments

Describe the bug
When scanning hugging face models, modelscan is not properly handling whitespaces and crashing.

To Reproduce
Steps to reproduce the behavior:

  1. pipe install modelscan
  2. modelscan -hf stabilityai/StableBeluga2

Expected behavior
modelscan should escape these URLs properly.

Screenshots
Screenshot 2023-08-08 at 9 06 22 AM

Environment (please complete the following information):

  • macOS-13.3.1-arm64-arm-64bit
  • Modelscan Version: latest
  • ML Framework version: Any

Additional context
N/A