Hugging Face's logo

Spaces:
ohayonguy
/
PMRF
Running on Zero

App Files Community
4
ohayonguy commited on
Commit
9a668a2
•
1 Parent(s): 6fe04b4

test

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=device, dtype=torch.bfloat16)
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=device)
33
 
34
  face_helper_dummy = FaceRestoreHelper(
35
  1,