Spaces:

takahirox
/

text_to_texture

Runtime error

takahirox commited on Dec 15, 2023

Commit

cdd743a

•

1 Parent(s): 62f266f

WIP commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,10 +7,12 @@ model_id = "dream-textures/texture-diffusion"
 device = "cuda"
 dtype = torch.float16
-pipe = StableDiffusionPipeline.from_pretrained(
-  model_id, torch_dtype=dtype
-).to(device)
-#pipe = StableDiffusionPipeline.from_pretrained(model_id)
 compel_proc = Compel(
   tokenizer=pipe.tokenizer,

 device = "cuda"
 dtype = torch.float16
+if torch.cuda.is_available():
+  pipe = StableDiffusionPipeline.from_pretrained(
+    model_id, torch_dtype=dtype
+  ).to(device)
+else:
+  pipe = StableDiffusionPipeline.from_pretrained(model_id)
 compel_proc = Compel(
   tokenizer=pipe.tokenizer,