Update fp8.py
This commit is contained in:
parent
cb00618e49
commit
db292872c8
1 changed files with 1 additions and 0 deletions
|
|
@ -566,6 +566,7 @@ try:
|
|||
logger.info(f"Using fbgemm_gpu block quantized FP8 matmul")
|
||||
fp8_block_quant_linear = fp8_fbgemm_block_linear
|
||||
else:
|
||||
os.environ["UNSLOTH_HAS_FBGEMM"] = "0"
|
||||
except:
|
||||
pass
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue