add widget

2023-12-08 09:41:35 +01:00
2 changed files with 4 additions and 6 deletions
@@ -157,6 +157,8 @@ tags:
 base_model: {base_model}
 instance_prompt: {instance_prompt}
 license: openrail++
+widget:
+    - text: '{validation_prompt if validation_prompt else instance_prompt}'
 ---
 """

@@ -820,9 +820,7 @@ def _is_torch_fp16_available(device):

    try:
        x = torch.zeros((2, 2), dtype=torch.float16).to(device)
-        _ = torch.mul(x, x)
-        return True
-
+        _ = x @ x
    except Exception as e:
        if device.type == "cuda":
            raise ValueError(
@@ -840,9 +838,7 @@ def _is_torch_fp64_available(device):

    try:
        x = torch.zeros((2, 2), dtype=torch.float64).to(device)
-        _ = torch.mul(x, x)
-        return True
-
+        _ = x @ x
    except Exception as e:
        if device.type == "cuda":
            raise ValueError(