Fetching metadata from the HF Docker repository...
Update app.py
dad4b71 verified - 1.52 kB initial commit
- 2.22 kB Update README.md
- 2.36 kB Update app.py
imdb_sentiment_lora_model.pkl Detected Pickle imports (12)
- "torch.storage._load_from_bytes",
- "peft.utils.peft_types.TaskType",
- "peft.tuners.lora.config.LoraRuntimeConfig",
- "tokenizers.Tokenizer",
- "tokenizers.models.Model",
- "peft.tuners.lora.config.LoraConfig",
- "transformers.models.distilbert.tokenization_distilbert_fast.DistilBertTokenizerFast",
- "transformers.models.distilbert.configuration_distilbert.DistilBertConfig",
- "torch._utils._rebuild_tensor_v2",
- "peft.utils.peft_types.PeftType",
- "collections.OrderedDict",
- "torch.float32"
How to fix it?
268 MB Upload imdb_sentiment_lora_model.pkl imdb_sentiment_lora_model_cpu.pt Detected Pickle imports (12)
- "tokenizers.Tokenizer",
- "peft.tuners.lora.config.LoraRuntimeConfig",
- "transformers.models.distilbert.configuration_distilbert.DistilBertConfig",
- "peft.tuners.lora.config.LoraConfig",
- "tokenizers.models.Model",
- "torch.float32",
- "torch.FloatStorage",
- "peft.utils.peft_types.TaskType",
- "peft.utils.peft_types.PeftType",
- "transformers.models.distilbert.tokenization_distilbert_fast.DistilBertTokenizerFast",
- "torch._utils._rebuild_tensor_v2",
- "collections.OrderedDict"
How to fix it?
268 MB Upload imdb_sentiment_lora_model_cpu.pt - 81 Bytes Update requirements.txt