Spaces:

BoHuangLab
/

CELL-FM

Running on Zero

CELL-FM CondenSeq demo: sequence -> titration curve, AUC and AAC

e794567 verified about 2 months ago

411 Bytes

# -*- coding: utf-8 -*-
import os

import torch


def is_master_node():
    if "RANK" not in os.environ or int(os.environ["RANK"]) == 0:
        return True
    else:
        return False


def support_flash_attntion():
    # check the GPU is A100 or H100
    device_name = torch.cuda.get_device_name(0)
    for name in ["A100", "H100"]:
        if name in device_name:
            return True
    return False

	# -- coding: utf-8 --
	import os

	import torch


	def is_master_node():
	if "RANK" not in os.environ or int(os.environ["RANK"]) == 0:
	return True
	else:
	return False


	def support_flash_attntion():
	# check the GPU is A100 or H100
	device_name = torch.cuda.get_device_name(0)
	for name in ["A100", "H100"]:
	if name in device_name:
	return True
	return False