update

Merge branch 'main' into compile-test-fix
update
2023-12-08 12:28:50 +00:00 · 2023-12-08 11:10:18 +00:00 · 2023-12-08 09:58:13 +00:00
2 changed files with 6 additions and 4 deletions
@@ -157,8 +157,6 @@ tags:
 base_model: {base_model}
 instance_prompt: {instance_prompt}
 license: openrail++
-widget:
-    - text: '{validation_prompt if validation_prompt else instance_prompt}'
 ---
 """

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

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

    try:
        x = torch.zeros((2, 2), dtype=torch.float64).to(device)
-        _ = x @ x
+        _ = torch.mul(x, x)
+        return True
+
    except Exception as e:
        if device.type == "cuda":
            raise ValueError(
Author	SHA1	Message	Date
Dhruv Nair	644846aa14	update	2023-12-08 12:28:50 +00:00
Dhruv Nair	6a90d8a023	Merge branch 'main' into compile-test-fix	2023-12-08 11:10:18 +00:00
Dhruv Nair	7270cbf421	update	2023-12-08 09:58:13 +00:00