Insta360-Research
/

DiT360-Panorama-Image-Generation

Model card Files Files and versions

Insta360-Research commited on 26 days ago

Commit

945bfe6

·

verified ·

1 Parent(s): c0b8bd5

Update inference.py

Files changed (1) hide show

inference.py +1 -1

inference.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 device = torch.device("cuda:0")
 pipe = DiT360Pipeline.from_pretrained("black-forest-labs/FLUX.1-dev", dtype=torch.float16).to(device)
-pipe.load_lora_weights("fenghora/DiT360-Panorama-Image-Generation")
 image = pipe(
     "This is a panorama. The image shows a medieval castle stands proudly on a hilltop surrounded by autumn forests, with golden light spilling across the landscape.",

 device = torch.device("cuda:0")
 pipe = DiT360Pipeline.from_pretrained("black-forest-labs/FLUX.1-dev", dtype=torch.float16).to(device)
+pipe.load_lora_weights("Insta360-Research/DiT360-Panorama-Image-Generation")
 image = pipe(
     "This is a panorama. The image shows a medieval castle stands proudly on a hilltop surrounded by autumn forests, with golden light spilling across the landscape.",