Spaces:

Insta360-Research
/

DiT360

Running on Zero

Insta360-Research commited on Oct 11

Commit

ee241e0

verified ·

1 Parent(s): 8495f21

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 model_repo = "black-forest-labs/FLUX.1-dev"
-lora_weights = "fenghora/DiT360-Panorama-Image-Generation"
 pipe = DiT360Pipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
 pipe.load_lora_weights(lora_weights)

 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 model_repo = "black-forest-labs/FLUX.1-dev"
+lora_weights = "Insta360-Research/DiT360-Panorama-Image-Generation"
 pipe = DiT360Pipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
 pipe.load_lora_weights(lora_weights)