Spaces:

Frederic-CellNum
/

api_sparrow_ocr

Sleeping

Create download_model.py

82aed82 verified 8 months ago

367 Bytes

from transformers import Qwen2VLForConditionalGeneration, AutoProcessor

print("Téléchargement du processor...")
AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")

print("Téléchargement du modèle...")
Qwen2VLForConditionalGeneration.from_pretrained(
    "Qwen/Qwen2-VL-7B-Instruct",
    torch_dtype="auto"
)

print("Modèle téléchargé avec succès!")

	from transformers import Qwen2VLForConditionalGeneration, AutoProcessor

	print("Téléchargement du processor...")
	AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")

	print("Téléchargement du modèle...")
	Qwen2VLForConditionalGeneration.from_pretrained(
	"Qwen/Qwen2-VL-7B-Instruct",
	torch_dtype="auto"
	)

	print("Modèle téléchargé avec succès!")