Spaces:

codewithdark
/

Faceless-video

Running

codewithdark commited on Sep 14, 2024

Commit

95a8cb6

verified ·

1 Parent(s): ab9f5d8

Update utility/image_generator.py

Files changed (1) hide show

utility/image_generator.py CHANGED Viewed

@@ -12,7 +12,7 @@ load_dotenv()
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipeline = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
 pipeline = pipeline.to(device)
 def generate_image_prompts(script):

 device = "cuda" if torch.cuda.is_available() else "cpu"
+pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0")
 pipeline = pipeline.to(device)
 def generate_image_prompts(script):