Aracılığıyla paylaş


Temel Model API'lerini kullanarak toplu çıkarım

Bu makalede, Temel Model API'lerini kullanarak sağlanan aktarım hızı uç noktasında toplu çıkarım gerçekleştiren örnek not defterleri sağlanır. Temel Model API'lerini kullanarak toplu çıkarım gerçekleştirmek için her iki not defterine de ihtiyacınız vardır.

Örneklerde sohbet görevleri için DBRX Instruct modeli kullanılarak toplu çıkarım gösterilmektedir.

Gereksinimler

  • Temel Model API'lerinde desteklenen bir çalışma alanı
  • Databricks Runtime 14.0 ML veya üzeri
  • Not provisioned-throughput-batch-inference defteri ve chat-batch-inference-api not defteri çalışma alanında aynı dizinde bulunmalıdır

Giriş tablosunu ayarlama, toplu çıkarım

Aşağıdaki not defteri Python kullanarak aşağıdaki görevleri yerine getirmez:

  • Giriş tablosundan ve giriş sütunundan verileri okur
  • İstekleri oluşturur ve bir Temel Model API'leri uç noktasına gönderir
  • Giriş satırlarını yanıt verileriyle birlikte çıkış tablosuna kalıcı hale getiren

Python not defterini kullanarak sohbet modeli toplu çıkarım görevleri

Not defterini alma

Aşağıdaki not defteri yukarıdaki not defteriyle aynı görevleri yerine getirmesine karşın Spark'ı kullanıyor:

  • Giriş tablosundan ve giriş sütunundan verileri okur
  • İstekleri oluşturur ve bir Temel Model API'leri uç noktasına gönderir
  • Giriş satırını yanıt verileriyle birlikte çıkış tablosuna kalıcı hale getiren

PySpark Pandas UDF not defterini kullanarak sohbet modeli toplu çıkarım görevleri

Not defterini alma

Sağlanan aktarım hızı uç noktası oluşturma

Python not defteri yerine Spark not defterini kullanmak istiyorsanız Python not defterini çağıran komutu güncelleştirdiğinizden emin olun.

  • Sağlanan aktarım hızı sunan uç nokta oluşturur
  • Uç noktayı hazır duruma gelene kadar izleme
  • chat-batch-inference-api Hazırlanan uç noktada toplu çıkarım görevlerini eşzamanlı olarak çalıştırmak için not defterini çağırır. Spark'ı kullanmayı tercih ediyorsanız, bu başvuruyu not defterini çağırmak chat-batch-inference-udf için değiştirin.
  • Toplu çıkarım tamamlandıktan sonra sağlanan aktarım hızı sunan uç noktayı siler

Sağlanan aktarım hızı uç noktası not defterinde toplu çıkarım gerçekleştirme

Not defterini alma

Ek kaynaklar