Hugging Face's logo

Spaces:
feishen29
/
IMAGDressing-v1
Running on Zero

App Files Community
1
feishen29 commited on
Commit
0132256
•
1 Parent(s): 17eb228

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ args = parser.parse_args()
74
 
75
  args.device = "cuda"
76
 
77
- # base_path = 'feishen29/IMAGDressing-v1'
78
 
79
 
80
  vae = AutoencoderKL.from_pretrained('stabilityai/sd-vae-ft-mse').to(dtype=torch.float16, device=args.device)
 
74
 
75
  args.device = "cuda"
76
 
77
+ base_path = 'feishen29/IMAGDressing-v1'
78
 
79
 
80
  vae = AutoencoderKL.from_pretrained('stabilityai/sd-vae-ft-mse').to(dtype=torch.float16, device=args.device)