Hugging Face's logo

Spaces:
runito
/
DiT360
Sleeping

App Files Community
Insta360-Research commited on
Commit
bbff29a
·
verified ·
1 Parent(s): 3025733

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ if torch.cuda.is_available():
12
  else:
13
  torch_dtype = torch.float32
14
 
15
- model_repo = "/media/nfs/tmp_data/fenghr/download/FLUX.1-dev"
16
- lora_weights = "/media/nfs/tmp_data/fenghr/DiT360-FLUX-Panorama-Lora"
17
 
18
  pipe = DiT360Pipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
19
  pipe.load_lora_weights(lora_weights)
 
12
  else:
13
  torch_dtype = torch.float32
14
 
15
+ model_repo = "black-forest-labs/FLUX.1-dev"
16
+ lora_weights = "fenghora/DiT360-Panorama-Image-Generation"
17
 
18
  pipe = DiT360Pipeline.from_pretrained(model_repo, torch_dtype=torch_dtype).to(device)
19
  pipe.load_lora_weights(lora_weights)