deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

Merge pull request #77 from Cyberes/main

Browse Source 
Allow hiding of the welcome message
This commit is contained in:
Tim Dettmers 2023-01-02 11:28:17 +01:00 committed by GitHub
commit df9a9b0c4c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bitsandbytes/cuda_setup/main.py
View File

@ -17,6 +17,7 @@ evaluation:
""" """
import ctypes import ctypes
import os
import torch import torch
@ -112,12 +113,12 @@ def get_compute_capability(cuda):
def evaluate_cuda_setup(): def evaluate_cuda_setup():
# we remove this for now and see how things go if 'BITSANDBYTES_NOWELCOME' not in os.environ or str(os.environ['BITSANDBYTES_NOWELCOME']) == '0':
#print('') print('')
#print('='*35 + 'BUG REPORT' + '='*35) print('='*35 + 'BUG REPORT' + '='*35)
#print('Welcome to bitsandbytes. For bug reports, please submit your error trace to: https://github.com/TimDettmers/bitsandbytes/issues') print('Welcome to bitsandbytes. For bug reports, please submit your error trace to: https://github.com/TimDettmers/bitsandbytes/issues')
#print('For effortless bug reporting copy-paste your error into this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link') print('For effortless bug reporting copy-paste your error into this form: https://docs.google.com/forms/d/e/1FAIpQLScPB8emS3Thkp66nvqwmjTEgxp8Y9ufuWTzFyr9kJ5AoI47dQ/viewform?usp=sf_link')
#print('='*80) print('='*80)
if not torch.cuda.is_available(): return 'libsbitsandbytes_cpu.so', None, None, None, None if not torch.cuda.is_available(): return 'libsbitsandbytes_cpu.so', None, None, None, None
cuda_setup = CUDASetup.get_instance() cuda_setup = CUDASetup.get_instance()

1
environment.yml
View File

@ -1,6 +1,7 @@
name: 8-bit name: 8-bit
channels: channels:
- conda-forge - conda-forge
- pytorch
dependencies: dependencies:
- python=3.9 - python=3.9
- pytest - pytest