rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-05-28 09:20:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

isort entire repo (#200)

Browse Source
This commit is contained in:
Rob Meng
2023-06-15 20:12:10 -04:00
committed by GitHub
parent 01dadb0862
commit d1e8a97a2a
15 changed files with 29 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/src/notebooks/diffusiondb/datagen.py
View File

@@ -21,12 +21,13 @@ from argparse import ArgumentParser
from multiprocessing import Pool
import lance
import lancedb
import pyarrow as pa
from datasets import load_dataset
from PIL import Image
from transformers import CLIPModel, CLIPProcessor, CLIPTokenizerFast
import lancedb
MODEL_ID = "openai/clip-vit-base-patch32"
device = "cuda"