Instructions to use EndeavourDD/gnn_wm with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use EndeavourDD/gnn_wm with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("EndeavourDD/gnn_wm", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
| {"frame_idx": 0, "nodes": [{"id": "cpu_fan", "type": "cpu_fan", "color": "#FF6B6B", "visible": true, "centroid_3d": [-0.003319174749776721, -0.05959661304950714, 0.6469275951385498], "embedding_norm": 2.339418649673462, "mask_area": 25497}, {"id": "graphic_card", "type": "graphic_card", "color": "#FF8C42", "visible": true, "centroid_3d": [-0.14235544204711914, -0.10416606813669205, 0.637933075428009], "embedding_norm": 2.049105167388916, "mask_area": 14849}, {"id": "connector_1", "type": "connector", "color": "#DDA0DD", "visible": true, "centroid_3d": [-0.07623299211263657, 0.019043894484639168, 0.7628206610679626], "embedding_norm": 3.3849947452545166, "mask_area": 406}, {"id": "connector_2", "type": "connector", "color": "#DDA0DD", "visible": true, "centroid_3d": [-0.03202902525663376, 0.015756305307149887, 0.7530792355537415], "embedding_norm": 3.219653606414795, "mask_area": 476}, {"id": "connector_3", "type": "connector", "color": "#DDA0DD", "visible": true, "centroid_3d": [0.018211210146546364, 0.019433297216892242, 0.75971919298172], "embedding_norm": 4.50897216796875, "mask_area": 473}, {"id": "connector_4", "type": "connector", "color": "#DDA0DD", "visible": true, "centroid_3d": [0.05329165607690811, -0.14644543826580048, 0.6421629786491394], "embedding_norm": 2.9468390941619873, "mask_area": 530}, {"id": "ram_1", "type": "ram", "color": "#FFEAA7", "visible": true, "centroid_3d": [-0.0030889410991221666, -0.001728980801999569, 0.731830358505249], "embedding_norm": 3.054723024368286, "mask_area": 4679}, {"id": "cpu_bracket", "type": "cpu_bracket", "color": "#4ECDC4", "visible": false, "centroid_3d": [0, 0, 0], "embedding_norm": 0.0, "mask_area": 0}, {"id": "cpu", "type": "cpu", "color": "#45B7D1", "visible": false, "centroid_3d": [0, 0, 0], "embedding_norm": 0.0, "mask_area": 0}, {"id": "motherboard", "type": "motherboard", "color": "#8B5CF6", "visible": true, "centroid_3d": [-0.07307398319244385, -0.09350324422121048, 0.6752018332481384], "embedding_norm": 1.7568132877349854, "mask_area": 155739}, {"id": "ram_clip_1", "type": "ram_clip", "color": "#96CEB4", "visible": true, "centroid_3d": [0.07286746799945831, -0.0005528790643438697, 0.7657496929168701], "embedding_norm": 3.793574571609497, "mask_area": 188}, {"id": "ram_clip_2", "type": "ram_clip", "color": "#96CEB4", "visible": true, "centroid_3d": [-0.07836821675300598, -0.00022328570776153356, 0.7520178556442261], "embedding_norm": 3.4605636596679688, "mask_area": 223}, {"id": "ram_2", "type": "ram", "color": "#FFEAA7", "visible": true, "centroid_3d": [-0.008224790915846825, -0.017786994576454163, 0.700673520565033], "embedding_norm": 2.9754316806793213, "mask_area": 1264}, {"id": "ram_clip_3", "type": "ram_clip", "color": "#96CEB4", "visible": true, "centroid_3d": [-0.07728748023509979, -0.01995541900396347, 0.7257604002952576], "embedding_norm": 0.0, "mask_area": 142}, {"id": "ram_clip_4", "type": "ram_clip", "color": "#96CEB4", "visible": true, "centroid_3d": [0.07267742604017258, -0.020245257765054703, 0.7475025057792664], "embedding_norm": 3.968529224395752, "mask_area": 189}, {"id": "robot", "type": "robot", "color": "#F5F5F5", "visible": true, "centroid_3d": [-0.16381078958511353, 0.07366017252206802, 0.33944249153137207], "embedding_norm": 2.22784686088562, "mask_area": 22013}], "edges": [{"src": "cpu_bracket", "dst": "cpu", "is_locked": true}, {"src": "cpu_bracket", "dst": "motherboard", "is_locked": true}, {"src": "cpu", "dst": "motherboard", "is_locked": true}, {"src": "graphic_card", "dst": "motherboard", "is_locked": true}, {"src": "connector_1", "dst": "motherboard", "is_locked": true}, {"src": "connector_2", "dst": "motherboard", "is_locked": true}, {"src": "connector_3", "dst": "motherboard", "is_locked": true}, {"src": "connector_4", "dst": "motherboard", "is_locked": true}, {"src": "ram_1", "dst": "motherboard", "is_locked": true}, {"src": "ram_2", "dst": "motherboard", "is_locked": true}, {"src": "ram_clip_2", "dst": "ram_1", "is_locked": true}, {"src": "ram_clip_1", "dst": "ram_1", "is_locked": true}, {"src": "ram_clip_4", "dst": "ram_2", "is_locked": true}, {"src": "ram_clip_3", "dst": "ram_2", "is_locked": true}]} |