space-nuko / a1111-stable-diffusion-webui-vram-estimator

Show estimated VRAM usage for generation configs
MIT License
107 stars 11 forks source link

Error on Apple M1 #13

Open gzjaychen opened 1 year ago

gzjaychen commented 1 year ago

return torch._C._cuda_resetPeakMemoryStats(device) AttributeError: module 'torch._C' has no attribute '_cuda_resetPeakMemoryStats' [VRAMEstimator] Starting benchmark... Traceback (most recent call last): File "/Users/xxx/github/stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/routes.py", line 422, in run_predict output = await app.get_blocks().process_api( File "/Users/xxx/github/stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/blocks.py", line 1323, in process_api result = await self.call_function( File "/Users/xxx/github/stable-diffusion-webui/venv/lib/python3.10/site-packages/gradio/blocks.py", line 1051, in call_function prediction = await anyio.to_thread.run_sync( File "/Users/xxx/github/stable-diffusion-webui/venv/lib/python3.10/site-packages/anyio/to_thread.py", line 33, in run_sync return await get_asynclib().run_sync_in_worker_thread( File "/Users/xxx/github/stable-diffusion-webui/venv/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 877, in run_sync_in_worker_thread return await future File "/Users/xxx/github/stable-diffusion-webui/venv/lib/python3.10/site-packages/anyio/_backends/_asyncio.py", line 807, in run result = context.run(func, *args) File "/Users/xxx/github/stable-diffusion-webui/extensions/a1111-stable-diffusion-webui-vram-estimator/scripts/vram_estimator.py", line 119, in run_benchmark mem_stats = get_memory_stats() File "/Users/xxx/github/stable-diffusion-webui/extensions/a1111-stable-diffusion-webui-vram-estimator/scripts/vram_estimator.py", line 92, in get_memory_stats torch.cuda.reset_peak_memory_stats() File "/Users/xxx/github/stable-diffusion-webui/venv/lib/python3.10/site-packages/torch/cuda/memory.py", line 281, in reset_peak_memory_stats