Fabrice-TIERCELIN
commited on
Commit
•
948fcf4
1
Parent(s):
811f1d8
Activate
Browse files- GenVideo_app.py +1 -1
GenVideo_app.py
CHANGED
@@ -65,7 +65,7 @@ if torch.cuda.device_count() > 0:
|
|
65 |
model = model.to(SUPIR_device)
|
66 |
model.first_stage_model.denoise_encoder_s1 = copy.deepcopy(model.first_stage_model.denoise_encoder)
|
67 |
model.current_model = 'v0-Q'
|
68 |
-
|
69 |
|
70 |
# load LLaVA
|
71 |
#if use_llava:
|
|
|
65 |
model = model.to(SUPIR_device)
|
66 |
model.first_stage_model.denoise_encoder_s1 = copy.deepcopy(model.first_stage_model.denoise_encoder)
|
67 |
model.current_model = 'v0-Q'
|
68 |
+
ckpt_Q, ckpt_F = load_QF_ckpt(args.opt)
|
69 |
|
70 |
# load LLaVA
|
71 |
#if use_llava:
|