[deepfloydif.py] made spelling consistent (deepfloydif, not deepfloyd)

deepfloydif.py

@@ -11,12 +11,12 @@ import glob
 import pathlib
 
 HF_TOKEN = os.getenv('HF_TOKEN')
-deepfloyd_client = Client("huggingface-projects/IF", HF_TOKEN)
+deepfloydif_client = Client("huggingface-projects/IF", HF_TOKEN)
 
 BOT_USER_ID = 1086256910572986469 if os.getenv("TEST_ENV", False) else 1102236653545861151
-DEEPFLOYD_CHANNEL_ID = 1121834257959092234 if os.getenv("TEST_ENV", False) else 1119313215675973714
+DEEPFLOYDIF_CHANNEL_ID = 1121834257959092234 if os.getenv("TEST_ENV", False) else 1119313215675973714
 
-def deepfloyd_stage_1_inference(prompt):
+def deepfloydif_stage_1_inference(prompt):
     """Generates an image based on a prompt"""
     negative_prompt = ''
     seed = random.randint(0, 1000)
@@ -34,23 +34,23 @@ def deepfloyd_stage_1_inference(prompt):
         stage_1_result_path = result[2]
         return [stage_1_results, stage_1_result_path]     
 
-def deepfloyd_stage_2_inference(index, stage_1_result_path):
-    """Upscales one of the images from deepfloyd_stage_1_inference based on the chosen index"""
+def deepfloydif_stage_2_inference(index, stage_1_result_path):
+    """Upscales one of the images from deepfloydif_stage_1_inference based on the chosen index"""
     selected_index_for_stage_2 = index
     seed_2 = 0
     guidance_scale_2 = 4
     custom_timesteps_2 = 'smart50'
     number_of_inference_steps_2 = 50
-    result_path = deepfloyd_client.predict(stage_1_result_path, selected_index_for_stage_2, seed_2, guidance_scale_2, custom_timesteps_2, number_of_inference_steps_2, api_name='/upscale256')
+    result_path = deepfloydif_client.predict(stage_1_result_path, selected_index_for_stage_2, seed_2, guidance_scale_2, custom_timesteps_2, number_of_inference_steps_2, api_name='/upscale256')
     return result_path 
     
 async def deepfloydif_stage_1(interaction, prompt, client):
     """DeepfloydIF command (generate images with realistic text using slash commands)"""
     try:
         global BOT_USER_ID
-        global DEEPFLOYD_CHANNEL_ID
+        global DEEPFLOYDIF_CHANNEL_ID
         if interaction.user.id != BOT_USER_ID:
-            if interaction.channel.id == DEEPFLOYD_CHANNEL_ID:
+            if interaction.channel.id == DEEPFLOYDIF_CHANNEL_ID:
                 await interaction.response.send_message("Working on it!")
                 channel = interaction.channel
                 # interaction.response message can't be used to create a thread, so we create another message
@@ -62,7 +62,7 @@ async def deepfloydif_stage_1(interaction, prompt, client):
 
                 '''
                 loop = asyncio.get_running_loop()
-                result = await loop.run_in_executor(None, deepfloyd_stage_1_inference, prompt)  
+                result = await loop.run_in_executor(None, deepfloydif_stage_1_inference, prompt)  
                 stage_1_results = result[0]
                 stage_1_result_path = result[2]                
                 '''
@@ -134,7 +134,7 @@ async def deepfloydif_stage_2(index: int, stage_1_result_path, thread, dfif_comm
         
         # run blocking function in executor
         loop = asyncio.get_running_loop()
-        result_path = await loop.run_in_executor(None, deepfloyd_stage_2_inference, index, stage_1_result_path) 
+        result_path = await loop.run_in_executor(None, deepfloydif_stage_2_inference, index, stage_1_result_path) 
         
         with open(result_path, 'rb') as f:
             await thread.send('Here is the upscaled image!', file=discord.File(f, 'result.png'))
@@ -147,11 +147,11 @@ async def deepfloydif_stage_2_react_check(reaction, user):
     """Checks for a reaction in order to call dfif2"""
     try:
         global BOT_USER_ID
-        global DEEPFLOYD_CHANNEL_ID
+        global DEEPFLOYDIF_CHANNEL_ID
         if user.id != BOT_USER_ID:
             thread = reaction.message.channel
             thread_parent_id = thread.parent.id
-            if thread_parent_id == DEEPFLOYD_CHANNEL_ID: 
+            if thread_parent_id == DEEPFLOYDIF_CHANNEL_ID: 
                 if reaction.message.attachments:
                     if user.id == reaction.message.mentions[0].id: 
                         attachment = reaction.message.attachments[0]