Bắt đầu ngayBắt đầu miễn phí

Tìm mô hình text-to-image phổ biến nhất

Đến lúc tinh chỉnh tìm kiếm để tìm và tải mô hình Stable Diffusion text-to-image phổ biến nhất từ CompVis trên Hugging Face Hub.

Photo of a labrador chasing a ball

API của Hugging Face Hub (api) đã được tải, cùng với mô-đun StableDiffusionPipeline từ thư viện diffusers. Với các mô hình Stable Diffusion, bạn có thể tạo ảnh từ bất kỳ prompt nào bạn muốn, ví dụ: "a black labrador chasing a tennis ball".

Bài tập này là một phần của khóa học

Mô hình đa phương thức với Hugging Face

Xem khóa học

Hướng dẫn bài tập

  • Chỉ sử dụng các mô hình cho tác vụ text-to-image.
  • Dùng thẻ (tag) phù hợp để đảm bảo mô hình có thể được tải bởi lớp StableDiffusionPipeline từ thư viện diffusers.
  • Sắp xếp kết quả theo số lượng "likes".
  • Tải mô hình phổ biến nhất từ models bằng ID của nó.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

models = api.list_models(
    # Filter for text-to-image tasks
    task="____",
    author="CompVis",
    # Filter for models that can be loaded by the diffusers library
    tags="diffusers:____",
    # Sort according to the most popular
    sort="____"
)

models = list(models)

# Load the most popular model from models
pipe = StableDiffusionPipeline.from_pretrained(models[0].____)
Chỉnh sửa và Chạy Mã