import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("hvent90/point_navigation", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]Simple Point Navigation
This generates a path between two points in a 2D grid. Basically AGI.
Usage
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("hvent90/point_navigation", trust_remote_code=True)
trajectory = pipeline(start=[0, 0], target=[0.8, 0.8]).trajectories[0]
print(trajectory) # 32 waypoints from start to target
- Downloads last month
- 1
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support