Spaces:

sudo-ai
/

zero123plus-demo-space

Running on A10G

zero123plus-demo-space / download_checkpoints.py

eliphatfs

Add queue and docker cache.

d56acb1 8 months ago

No virus

539 Bytes

	import os
	import torch
	import urllib.request
	import huggingface_hub
	from diffusers import DiffusionPipeline


	if 'HF_TOKEN' in os.environ:
	huggingface_hub.login(os.environ['HF_TOKEN'])
	sam_checkpoint = "tmp/sam_vit_h_4b8939.pth"
	os.makedirs('tmp', exist_ok=True)
	urllib.request.urlretrieve(
	"https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
	sam_checkpoint
	)
	DiffusionPipeline.from_pretrained(
	"sudo-ai/zero123plus-v1.1", custom_pipeline="sudo-ai/zero123plus-pipeline",
	torch_dtype=torch.float16
	)