From 2a08f8bb36dfc4109ff7730abcfb299ff90e8405 Mon Sep 17 00:00:00 2001 From: "majiahui@haimaqingfan.com" Date: Wed, 12 Jul 2023 18:30:26 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85readme?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- flask_batch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/flask_batch.py b/flask_batch.py index 16f9e20..3708802 100644 --- a/flask_batch.py +++ b/flask_batch.py @@ -1,5 +1,5 @@ import os -os.environ["CUDA_VISIBLE_DEVICES"] = "0" +os.environ["CUDA_VISIBLE_DEVICES"] = "3" from flask import Flask, jsonify from flask import request from transformers import pipeline