Unsloth Studio blocks the model because training_args.bin is flagged as CAUTION

#2
by p3pp01 - opened

Ciao,
ho provando a utilizzare rizzoaiacademy/rizzo-pii-0.3B come modello di partenza per un nuovo training tramite Unsloth Studio, invece che il file di training dentro src/.
Durante il caricamento, Unsloth blocca completamente il modello mostrando il messaggio:
“Unsafe files detected”

Il file indicato è:
training_args.bin — CAUTION

Da quanto risulta, il blocco non riguarda model.safetensors, ma il file training_args.bin generato normalmente da Transformers Trainer. Unsloth interpreta però la segnalazione CAUTION dello scanner di Hugging Face come un unsafe file e non permette di proseguire.
Potreste verificare se training_args.bin è necessario nel repository del modello e, in caso contrario, rimuoverlo? Il modello dovrebbe continuare a funzionare normalmente tramite config.json, tokenizer e model.safetensors, evitando però il blocco preventivo di Unsloth Studio o su altri.

Non sto segnalando la presenza accertata di malware: sembra piuttosto un problema di compatibilità tra il file serializzato e la politica di sicurezza più restrittiva.

Allego lo screenshot del messaggio.

image

Grazie!

Sign up or log in to comment