Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ def check_text(prompt, negative=""):
|
|
| 27 |
|
| 28 |
DESCRIPTION = """"""
|
| 29 |
if not torch.cuda.is_available():
|
| 30 |
-
DESCRIPTION += "\n<p
|
| 31 |
|
| 32 |
MAX_SEED = np.iinfo(np.int32).max
|
| 33 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
|
|
@@ -41,7 +41,7 @@ NUM_IMAGES_PER_PROMPT = 1
|
|
| 41 |
|
| 42 |
if torch.cuda.is_available():
|
| 43 |
pipe = DiffusionPipeline.from_pretrained(
|
| 44 |
-
"
|
| 45 |
##"SG161222/RealVisXL_V3.0",
|
| 46 |
torch_dtype=torch.float16,
|
| 47 |
use_safetensors=True,
|
|
|
|
| 27 |
|
| 28 |
DESCRIPTION = """"""
|
| 29 |
if not torch.cuda.is_available():
|
| 30 |
+
DESCRIPTION += "\n<p>⚠️Running on CPU, This may not work on CPU.</p>"
|
| 31 |
|
| 32 |
MAX_SEED = np.iinfo(np.int32).max
|
| 33 |
CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
|
|
|
|
| 41 |
|
| 42 |
if torch.cuda.is_available():
|
| 43 |
pipe = DiffusionPipeline.from_pretrained(
|
| 44 |
+
"SG161222/RealVisXL_V1.02_Turbo",
|
| 45 |
##"SG161222/RealVisXL_V3.0",
|
| 46 |
torch_dtype=torch.float16,
|
| 47 |
use_safetensors=True,
|