Aekanun commited on
Commit
f9f5ef2
·
1 Parent(s): ac7ed3e
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ import os
19
  # ตั้งค่า env variables ทั้งหมดก่อน import torch
20
  os.environ['TORCH_LOGS'] = '+dynamo'
21
  os.environ['TORCHDYNAMO_VERBOSE'] = '1'
22
- os.environ['TORCH_INDUCTOR_BACKEND'] = 'ATEN' # เพิ่มบรรทัดนี้
23
  # os.environ['NVIDIA_VISIBLE_DEVICES'] = '' # อาจต้องตรวจสอบว่าจำเป็นไหม
24
 
25
  os.environ['TORCHDYNAMO_DEBUG'] = '1' # เพิ่มเพื่อดู debug info
 
19
  # ตั้งค่า env variables ทั้งหมดก่อน import torch
20
  os.environ['TORCH_LOGS'] = '+dynamo'
21
  os.environ['TORCHDYNAMO_VERBOSE'] = '1'
22
+ os.environ['TORCH_INDUCTOR_BACKEND'] = 'CUDA'
23
  # os.environ['NVIDIA_VISIBLE_DEVICES'] = '' # อาจต้องตรวจสอบว่าจำเป็นไหม
24
 
25
  os.environ['TORCHDYNAMO_DEBUG'] = '1' # เพิ่มเพื่อดู debug info