ohayonguy
commited on
Commit
·
fc56fc6
1
Parent(s):
3cd5bc5
test bfloat16
Browse files
app.py
CHANGED
|
@@ -29,7 +29,7 @@ model = SRVGGNetCompact(num_in_ch=3, num_out_ch=3, num_feat=64, num_conv=32, ups
|
|
| 29 |
half = True if torch.cuda.is_available() else False
|
| 30 |
upsampler = RealESRGANer(scale=4, model_path=realesr_model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
|
| 31 |
|
| 32 |
-
pmrf = MMSERectifiedFlow.from_pretrained('ohayonguy/PMRF_blind_face_image_restoration').to(device)
|
| 33 |
|
| 34 |
face_helper_dummy = FaceRestoreHelper(
|
| 35 |
1,
|
|
|
|
| 29 |
half = True if torch.cuda.is_available() else False
|
| 30 |
upsampler = RealESRGANer(scale=4, model_path=realesr_model_path, model=model, tile=0, tile_pad=10, pre_pad=0, half=half)
|
| 31 |
|
| 32 |
+
pmrf = MMSERectifiedFlow.from_pretrained('ohayonguy/PMRF_blind_face_image_restoration').to(device).to(torch.bfloat16)
|
| 33 |
|
| 34 |
face_helper_dummy = FaceRestoreHelper(
|
| 35 |
1,
|