poloclub/diffusiondb
Updated • 26.7k • 612
How to use NouRed/fine-tuned-git-diffusion with Transformers:
# Use a pipeline as a high-level helper
# Warning: Pipeline type "image-to-text" is no longer supported in transformers v5.
# You must load the model directly (see below) or downgrade to v4.x with:
# 'pip install "transformers<5.0.0'
from transformers import pipeline
pipe = pipeline("image-to-text", model="NouRed/fine-tuned-git-diffusion") # Load model directly
from transformers import AutoProcessor, AutoModelForImageTextToText
processor = AutoProcessor.from_pretrained("NouRed/fine-tuned-git-diffusion")
model = AutoModelForImageTextToText.from_pretrained("NouRed/fine-tuned-git-diffusion")# Load model directly
from transformers import AutoProcessor, AutoModelForImageTextToText
processor = AutoProcessor.from_pretrained("NouRed/fine-tuned-git-diffusion")
model = AutoModelForImageTextToText.from_pretrained("NouRed/fine-tuned-git-diffusion")
# Use a pipeline as a high-level helper # Warning: Pipeline type "image-to-text" is no longer supported in transformers v5. # You must load the model directly (see below) or downgrade to v4.x with: # 'pip install "transformers<5.0.0' from transformers import pipeline pipe = pipeline("image-to-text", model="NouRed/fine-tuned-git-diffusion")