Spaces:

prithivMLmods
/

Imagineo-Chat

Running on Zero

prithivMLmods commited on 10 days ago

Commit

b83fa58

verified ·

1 Parent(s): 772bd03

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -262,8 +262,7 @@ if torch.cuda.is_available():
         "SG161222/RealVisXL_V5.0_Lightning", # / SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning
         torch_dtype=torch.float16,
         use_safetensors=True,
-        add_watermarker=False,
-        variant="fp16"
     ).to(device)
     if ENABLE_CPU_OFFLOAD:

         "SG161222/RealVisXL_V5.0_Lightning", # / SG161222/RealVisXL_V4.0 / SG161222/RealVisXL_V4.0_Lightning
         torch_dtype=torch.float16,
         use_safetensors=True,
+        add_watermarker=False
     ).to(device)
     if ENABLE_CPU_OFFLOAD: